Update test.py

chenyang1999 · chenyang1999 · commit ff69ded889e8 · 2019-05-03T19:08:03.000+08:00
diff --git a/.gitattributes b/.gitattributes
@@ -1 +1,3 @@
 *.html linguist-language=py
+*.md linguist-language=py
+*.ipynb linguist-language=py
diff --git a/One Day One GAN/day13/cyclegan/readme.md b/One Day One GAN/day13/cyclegan/readme.md
@@ -14,7 +14,7 @@ Hi ,my name is Chen Yang ,I am a sophomore in ocean university of China .I do so
 
 大名鼎鼎的 cycleGAN 就不用我多介绍了吧,那个大家每一次一讲到深度学习,总有人会放出来的那张图片,马变斑马的那个,就是出自于大名鼎鼎的 cycleGAN 了
 
-![horse2zebra](./cyclegan/horse2zebra.gif)
+![horse2zebra](./horse2zebra.gif)
 
 ![68747470733a2f2f6a756e79616e7a2e6769746875622e696f2f4379636c6547414e2f696d616765732f7465617365725f686967685f7265732e6a7067](https://ws2.sinaimg.cn/large/006tNc79ly1g2n89418lmj31le0rlqi1.jpg)
 
diff --git a/One Day One GAN/day13/cyclegan/test.py b/One Day One GAN/day13/cyclegan/test.py
@@ -48,8 +48,7 @@
     # Upsample image
     with torch.no_grad():
         sr_image = generator(image_tensor).cpu()
-
     # Save image
     fn = image_path.split("/")[-1]
     img_grid = torch.cat((image_tensor.cpu(),sr_image), 3)
-    save_image(img_grid, f"images/outputs/sr-{opt.checkpoint_model[-5]}-{fn[:-4]}.tif",nrow=1,normalize=True)
+    save_image(img_grid, f"images/outputs/{fn}",nrow=1,normalize=True)
