| ''' | |
| Author: [egrt] | |
| Date: 2022-08-23 13:21:27 | |
| LastEditors: [egrt] | |
| LastEditTime: 2022-08-23 13:45:21 | |
| Description: | |
| ''' | |
| #--------------------------------------------------------------# | |
| # 对单张图片进行预测，运行结果保存在根目录 | |
| # 默认保存文件为results/predict_out/predict_srgan.png | |
| #--------------------------------------------------------------# | |
| from PIL import Image | |
| from HEAT import HEAT | |
| if __name__ == "__main__": | |
| heat = HEAT() | |
| #----------------------------# | |
| # 单张图片的保存路径 | |
| #----------------------------# | |
| save_path = "assets/test_out.jpg" | |
| while True: | |
| img = input('Input image filename:') | |
| try: | |
| image = Image.open(img) | |
| except: | |
| print('Open Error! Try again!') | |
| continue | |
| else: | |
| r_image = heat.detect_one_image(image) | |
| r_image.save(save_path) | |
| r_image.show() |