mhamilton723/STEGO

get masks

widedh opened this issue · 2 comments

widedh commented

Hello everyone,

How can I get the mask of output images, please? thank you

tanveer6715 commented

Yes you can get mask by changing model path in eval config file to your save checkpoint file and then run python eval_segmentation.py file.

  • 👍1
www545 commented

Error executing job with overrides: []
Traceback (most recent call last):
File "G:\STEGO\src\eval_segmentation.py", line 187, in my_app
plot_img = (prep_for_plot(saved_data["img"][img_num]) * 255).numpy().astype(np.uint8)
IndexError: index 260 is out of bounds for dimension 0 with size 260
how to get all the mask of output images ?