-
Notifications
You must be signed in to change notification settings - Fork 58
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Problem when use best_model_pascal_ResFCN.pth #24
Comments
I had the same error. I think is because the However my predictions are very bad I am not sure if I made a mistake here or somewhere else. |
Thank you very much, |
@ThanhNguyenFG @sirtris |
@ThanhNguyenFG @sirtris @tongpinmo I have found out the problem with this code. In the file applyOnImage.py you can see the line "pred_counts = int(model.predict(batch, method="counts").ravel()[0])" |
Hello,
After I ran the command
python main.py -image_path figures/test.png
-model_path checkpoints/best_model_pascal_ResFCN.pth
-model_name ResFCN
I got this error:
"Model: ResFCN - Dataset: pascal - Metric: mRMSE
Traceback (most recent call last):
File "main.py", line 45, in
main()
File "main.py", line 33, in main
applyOnImage.apply(args.image_path, args.model_name, args.model_path)
File "/content/LCFCN/applyOnImage.py", line 34, in apply
imsave(save_path, ut.combine_image_blobs(image_raw, pred_blobs))
File "/usr/local/lib/python3.6/dist-packages/skimage/io/_io.py", line 144, in imsave
return call_plugin('imsave', fname, arr, plugin=plugin, **plugin_args)
File "/usr/local/lib/python3.6/dist-packages/skimage/io/manage_plugins.py", line 210, in call_plugin
return func(*args, **kwargs)
File "/usr/local/lib/python3.6/dist-packages/imageio/core/functions.py", line 253, in imwrite
raise ValueError("Image must be 2D (grayscale, RGB, or RGBA).")
ValueError: Image must be 2D (grayscale, RGB, or RGBA)."
I almost try many images but I can't fix this error.
Did I miss something?
Thank you very much.
The text was updated successfully, but these errors were encountered: