giacomopiccinini/unsupervisedimagesegmentation

Unsupervised Image Segmentation using PyTorch

unsupervisedimagesegmentation's Issues

CUDA not used

Dear Why do you not use CUDA?

Segmentation only resulting one label

hi,
thank you for expanding the code for multiple images. I tried using my own datasets containing thin section of rock images where image size is various. However, the training run smoothly and stop at very early epoch due to the labels reach min number of labels.
do you have a suggestion on how to improve the result? the segmentation result contains only single color for every grid.

thank you
Regards,

How to use the pretrained model for inference?

Dear can you make an example of the use of the trained model with the main that can be used for inference?