icoz69 / CaNet

The code for paper "CANet: Class-Agnostic Segmentation Networks with Iterative Refinement and Attentive Few-Shot Learning"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

how to test the trained model?

maobj opened this issue · comments

commented

Thank you for releasing the code
But I want to know how to test the trained model? I cannot find the test data or related code.

hi the validation dataloader can be used to eval the model temporarily, where all images are of size [321,321]. For the final evaluation, you need to modify the code slightly using the raw size of the images, and you can optionally add a multi-scale test into it.