|
@@ -80,7 +80,7 @@ eval_dataset = pdrs.datasets.VOCDetection(
|
|
# 构建YOLOv3模型,使用DarkNet53作为backbone
|
|
# 构建YOLOv3模型,使用DarkNet53作为backbone
|
|
# 目前已支持的模型请参考:https://github.com/PaddlePaddle/PaddleRS/blob/develop/docs/apis/model_zoo.md
|
|
# 目前已支持的模型请参考:https://github.com/PaddlePaddle/PaddleRS/blob/develop/docs/apis/model_zoo.md
|
|
# 模型输入参数请参考:https://github.com/PaddlePaddle/PaddleRS/blob/develop/paddlers/tasks/object_detector.py
|
|
# 模型输入参数请参考:https://github.com/PaddlePaddle/PaddleRS/blob/develop/paddlers/tasks/object_detector.py
|
|
-model = pdrs.tasks.YOLOv3(
|
|
|
|
|
|
+model = pdrs.tasks.det.YOLOv3(
|
|
num_classes=len(train_dataset.labels), backbone='DarkNet53')
|
|
num_classes=len(train_dataset.labels), backbone='DarkNet53')
|
|
|
|
|
|
# 执行模型训练
|
|
# 执行模型训练
|