batch run in pycharm(not terminal)

[Wangcehao]

only change the def main,code as follow:

def main(): test_path = 'testdata' save_path = 'result' imgs = os.listdir(test_path) time_sum = 0 for file in imgs: # img_name = sys.argv[1] img_name = os.path.join(test_path,file) img = imageio.imread(img_name) ta = time.time() result = Ying_2017_CAIP(img) # result: ndarray (m,n,4) time_sum += (time.time()-ta)

    # plt.imshow(result)
    # plt.show()
    save_image = cv2.cvtColor(result, cv2.COLOR_RGB2BGR)
    cv2.imwrite('result/{}'.format(file), save_image)
print('time_mean:', (time_sum/len(imgs)*1000))

if name == 'main': main()