search

bigmb / Unet-Segmentation-Pytorch-Nest-of-Unets

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet
MIT License
1.87k stars 345 forks source link

Test Models on Test Image Folder #47

Closed PrinceEAdjei closed 3 years ago

PrinceEAdjei commented 3 years ago

I would like to test the trained models on my test image folder. It seems the code does not do that. Rather, some pre-selected images are iteratively evaluated per each epoch during training.

I would like to use the already trained model to test my Test dataset and generate segmentation maps.

bigmb commented 3 years ago

I don't think I am following your question. If you are talking about testing the model on the test folder, you can check the code in pytorch_run.py from line 418. Because from that part, the testing starts of the model. Before that, there is also a validation of the training images.

PrinceEAdjei commented 3 years ago

Thank you. I double-checked the code. Works fine now.

I appreciate so much!