Spaces:

tidalove
/

yolox

Sleeping

Ge Zheng commited on Jul 26, 2021

Commit

1e0e26e

1 Parent(s): a4c8b2d

fix openvino python demo (#169)

Files changed (1) hide show

demo/OpenVINO/python/openvino_inference.py CHANGED Viewed

@@ -150,7 +150,7 @@ def main():
                          conf=args.score_thr, class_names=COCO_CLASSES)
     mkdir(args.output_dir)
-    output_path = os.path.join(args.output_dir, args.image_path.split("/")[-1])
     cv2.imwrite(output_path, origin_img)

                          conf=args.score_thr, class_names=COCO_CLASSES)
     mkdir(args.output_dir)
+    output_path = os.path.join(args.output_dir, args.input.split("/")[-1])
     cv2.imwrite(output_path, origin_img)