Update predict_system.py

This commit is contained in:
dyning 2020-05-14 14:43:20 +08:00 committed by GitHub
parent 2a6f5046c1
commit 3b931056c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -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))))