zishuo/ascend_community_projects
1
0
Fork 0
mirror of https://github.com/Ascend/ascend_community_projects.git synced 2025-09-26 20:01:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

update AnimeGAN/eval.py.

Browse Source 
使用if替代assert，并添加错误提示

Signed-off-by: 贝狄威尔 <751387081@qq.com>
This commit is contained in:
贝狄威尔
2022-09-26 08:11:26 +00:00
committed by Gitee
parent aacd79eac6
commit 239ae0f671
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
AnimeGAN/eval.py
View File

@@ -36,7 +36,13 @@ def eval_metrics(gpu_path, npu_path):
# should ensure the files'order keeping the same as before.
gpu_imgs = sorted(glob.glob(os.path.join(gpu_path, '*.jpg')))
npu_imgs = sorted(glob.glob(os.path.join(npu_path, '*.jpg')))
assert len(gpu_imgs) == len(npu_imgs)
if len(gpu_imgs) != len(npu_imgs):
print("The number of pictures in the two folders does not match!Please check it.")
print('It may be caused by:')
print(" 1.The outputPath dir not being empty before running main.py.")
print(" 2.Comparing with wrong gpu dir.")
print(" 3.Some pictures not being processed successfully in npu.")
exit()
for index, img1_path in enumerate(gpu_imgs):
img1 = cv2.imread(img1_path)