OasisYang / Wild6D

MIT License
44 stars 0 forks source link

evaluate shape-prior #1

Closed Liuchongpei closed 1 year ago

Liuchongpei commented 1 year ago

@OasisYang Hi, thanks for your great work! I used the evaluation code to evaluate shape-prior on Wild6D and got the separate results of 5 categories. I average 5 categories' results to get the final result. However, it seems much better than that in the paper. I don't know if I did something wrong.

OasisYang commented 1 year ago

Hi @Liuchongpei, did you directly evaluate the shape-prior pretrained model on Wild6D testing data? In that case, you should obtain the per-category results similar to the following table,

image

You could obtain slightly different results since the testing set is further cleaned.

Liuchongpei commented 1 year ago

Yes, I used the pretrained model trained on CAMERA+REAL. SPD_on_Wild6D The above is what we got. The Mug and Camera are similar to yours. But others are much better. And it prints 'Not found the ground truth from ...' when evaluating the Mug and Laptop.

OasisYang commented 1 year ago

Sorry, I don't know what happened here. How did you obtain the shape-prior results? We follow their official codes to generate the estimation results and only evaluate them on Wild6D with our evaluation script by setting the --only_eval to True. In addition, did you visualize your results? Here is a visualization result, you could compare it with yours.

image
Liuchongpei commented 1 year ago

We only make some modifications on 'evaluate_wild6d.py'. 改动0 改动1 改动2 Here is our visualization result. results_bottle_0001_1_0000

OasisYang commented 1 year ago

@Liuchongpei Could you just extract the poses with the official Shape-prior repo and evaluate with our evaluation script to see if you can obtain similar results?

I'm not sure what causes this difference. I guess the mean shapes used in Shape-prior are different with ours. We select the shape templates from ShapeNet dataset, while Shape-prior estimates the mean shape via a pretrained autoencoder.

Liuchongpei commented 1 year ago

Ok,I will have a try later. Did you evaluate NOCS on Wild6D?