search

mhamilton723 / STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences
MIT License
712 stars 143 forks source link

I can't .imshow() with IndexError #41

Open FrongTH opened 2 years ago

FrongTH commented 2 years ago

This is my IndexError image

This is directory dataset folder with the both folder have 32 unique_img_name. image

I changed only 2 parameter in train_config.yml for own dataset like their recommendations. image

I checked the inside file that error like this I'm not sure about that image

All library base on environment.yml How to fix them?

rcoabi commented 2 years ago

Setting n_images to be less or equal the batch_size in the train_config.yml should fix the issue.

KennyChen880127 commented 1 year ago

Setting n_images to be less or equal the batch_size in the train_config.yml should fix the issue.

Hello sir! Would you tell me your setting about n_image and batch size? Beacuse I stuck this issuie a lone time...