-
Notifications
You must be signed in to change notification settings - Fork 68
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
Testing new Image #23
Comments
Dear @ZshahRA Sorry to uncomfortable showing result images please refer main.py line 247 to add image information to visdom. if args.visualize:
if train_config["loss"] == "Lovasz":
grid_outputs = torchvision.utils.make_grid(color_transform((save_est[0] > 0).cpu().data), nrow=6)
else:
grid_outputs = torchvision.utils.make_grid(
color_transform(save_est[0].unsqueeze(0).cpu().max(1)[1].data), nrow=6)
This is visualized the validation image. |
Thank you @HYOJINPARK,
but i had the following issue Traceback (most recent call last): |
Hello @HYOJINPARK ,
I successfully trained the ExtremeC3Net. Now I want to test new images to see the testing accuracy. I am not sure what to do next. I tried putting images in Link512DATA folder and run Test_model.py. But there is no option to show images results.
cv2.imwrite("%s.jpg" %idx,np.asarray(segmented))
fig, axes = plt.subplots(1, 2, figsize=(12, 10))
axes[0].imshow(image)
Your response would be highly appreciated. Thank you
The text was updated successfully, but these errors were encountered: