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

fix cv2.error when use PIL.Image.Image

Browse Source
This commit is contained in:
shiraha721 2024-06-11 20:48:47 +08:00
parent 0efdd2f0c4
commit 38bcce53c2
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ultralytics/engine/predictor.py
View File

@ -375,6 +375,9 @@ 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=""):