Update predict_system.py
This commit is contained in:
parent
2a6f5046c1
commit
3b931056c4
|
@ -137,6 +137,6 @@ if __name__ == "__main__":
|
||||||
os.makedirs(draw_img_save)
|
os.makedirs(draw_img_save)
|
||||||
cv2.imwrite(
|
cv2.imwrite(
|
||||||
os.path.join(draw_img_save, os.path.basename(image_file)),
|
os.path.join(draw_img_save, os.path.basename(image_file)),
|
||||||
draw_img[:, :, -1])
|
draw_img[:, :, ::-1])
|
||||||
print("The visualized image saved in {}".format(
|
print("The visualized image saved in {}".format(
|
||||||
os.path.join(draw_img_save, os.path.basename(image_file))))
|
os.path.join(draw_img_save, os.path.basename(image_file))))
|
||||||
|
|
Loading…
Reference in New Issue