提交 e8fb682f 编写于 作者: H haoyuying

revise transform and pascalvoc dataset

上级 7e6b370f
...@@ -4,6 +4,6 @@ import paddlehub as hub ...@@ -4,6 +4,6 @@ import paddlehub as hub
if __name__ == '__main__': if __name__ == '__main__':
place = paddle.CUDAPlace(0) place = paddle.CUDAPlace(0)
paddle.disable_static() paddle.disable_static()
model = model = hub.Module(name='yolov3_darknet53_pascalvoc', is_train=False) model = hub.Module(name='yolov3_darknet53_pascalvoc', is_train=False)
model.eval() model.eval()
model.predict(imgpath="4026.jpeg", filelist="/PATH/TO/JSON/FILE") model.predict(imgpath="4026.jpeg", filelist="/PATH/TO/JSON/FILE")
...@@ -22,8 +22,6 @@ import numpy as np ...@@ -22,8 +22,6 @@ import numpy as np
from paddlehub.env import DATA_HOME from paddlehub.env import DATA_HOME
from pycocotools.coco import COCO from pycocotools.coco import COCO
from paddlehub.process.transforms import DetectCatagory, ParseImages
class DetectCatagory: class DetectCatagory:
"""Load label name, id and map from detection dataset. """Load label name, id and map from detection dataset.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册