xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update predictor.py

Browse Source
This commit is contained in:
shiraha721 2024-06-11 21:52:36 +08:00 committed by GitHub
parent 38bcce53c2
commit 8fd13b1b2c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ultralytics/engine/predictor.py
View File

@ -345,7 +345,8 @@ class BasePredictor:
if self.args.show: if self.args.show:
self.show(str(p)) self.show(str(p))
if self.args.save: if self.args.save:
self.save_predicted_images(str(self.save_dir / p.name), frame)
self.save_predicted_images(str(self.save_dir / (p.name or "tmp.jpg")), frame)
return string return string
@ -375,9 +376,6 @@ class BasePredictor:
# Save images # Save images
else: else:
if save_path.split("/")[-1].startswith("predict"):
# havn't provide file name
save_path = save_path + "/tmp.jpg"
cv2.imwrite(save_path, im) cv2.imwrite(save_path, im)
def show(self, p=""): def show(self, p=""):