Browse Source

Bug fix

tags/v1.2.10
deeppppp 2 years ago
parent
commit
c13016afe7
1 changed files with 2 additions and 2 deletions
  1. 2
    2
      utils.py

+ 2
- 2
utils.py View File

@@ -74,8 +74,8 @@ def check_shape(path, shape=Conf.desired_shape):
img_shape = imageio.mimread(path)[0][:, :, :3].shape

if img_shape != shape:
Conf.log.error("Image is not 512 x 512, got shape: {}".format(img_shape))
Conf.log.error("You should use one of the rescale options")
Conf.log.error("{} Image is not 512 x 512, got shape: {}".format(path, img_shape))
Conf.log.error("You should use one of the rescale options or manually resize the image")
sys.exit(1)



Loading…
Cancel
Save