Skip to content

Commit 9c0d4fb

Browse files
fix openvino python demo (Megvii-BaseDetection#169)
* fix openvino python demo
1 parent 8cf70b8 commit 9c0d4fb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

demo/OpenVINO/python/openvino_inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def main():
150150
conf=args.score_thr, class_names=COCO_CLASSES)
151151

152152
mkdir(args.output_dir)
153-
output_path = os.path.join(args.output_dir, args.image_path.split("/")[-1])
153+
output_path = os.path.join(args.output_dir, args.input.split("/")[-1])
154154
cv2.imwrite(output_path, origin_img)
155155

156156

0 commit comments

Comments
 (0)