search

czq142857 / BAE-NET

The code for paper "BAE-NET: Branched Autoencoder for Shape Co-Segmentation".
Other
68 stars 13 forks source link

Error calculating IOU #2

Closed LDHGItU closed 4 years ago

LDHGItU commented 4 years ago

Hi: When I train in an unsupervised way, the training phase is normal, but I encounter an error when I test the IOU, which prompts "imseg object has no attribute test" and how to solve this problem Thanks

czq142857 commented 4 years ago

We only provided code for testing IOUs on one-shot training results, since in that case, we know exactly which branch will output which part. For unsupervised training, the testing is a bit different because we cannot know which branch will output which part until after training is completed.

You need to visualize the segmentation results for a few shapes, and then manually assign a semantic label to the corresponding branch of the network. Once you have labeled all the branches of the network, you can easily modify function "test_pcSeg" to get the IOUs.

Best, Zhiqin