diff --git a/hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/module.py b/hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/module.py index e41e95ddfdd553cf7b3a6a296c7ca3a6a98bda04..98712df2b55d6777e2c827a9d6104ed406e671f0 100644 --- a/hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/module.py +++ b/hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/module.py @@ -27,7 +27,7 @@ from yolov3_resnet34_coco2017.yolo_head import MultiClassNMS, YOLOv3Head "Baidu's YOLOv3 model for object detection with backbone ResNet34, trained with dataset coco2017.", author="paddlepaddle", author_email="paddle-dev@baidu.com") -class YOLOv3DarkNet53Coco2017(hub.Module): +class YOLOv3ResNet34Coco2017(hub.Module): def _initialize(self): self.default_pretrained_model_path = os.path.join( self.directory, "yolov3_resnet34_model")