mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-23 05:24:22 +08:00
fix cv2.error when use PIL.Image.Image
This commit is contained in:
parent
0efdd2f0c4
commit
38bcce53c2
@ -375,6 +375,9 @@ class BasePredictor:
|
||||
|
||||
# Save images
|
||||
else:
|
||||
if save_path.split("/")[-1].startswith("predict"):
|
||||
# havn't provide file name
|
||||
save_path = save_path + "/tmp.jpg"
|
||||
cv2.imwrite(save_path, im)
|
||||
|
||||
def show(self, p=""):
|
||||
|
Loading…
x
Reference in New Issue
Block a user