twke18 / CAST

MIT License
29 stars 0 forks source link

code #4

Closed CurryKd7 closed 6 months ago

CurryKd7 commented 6 months ago

Hello, I trained train_cast.py on the ade20k dataset and obtained a checkpoint. pth model. Which. py file in the project should I run to check the image segmentation effect? I want to see the segmentation effect, not the classification

twke18 commented 6 months ago

Did you train image recognition on ade20K and would like to check the hierarchical segmentation results? You can check this ipython notebook.

If you fine-tuned the model on semantic segmentation in ade20K, this script would generate segmentation results.