search

Colin97 / OpenShape_code

official code of “OpenShape: Scaling Up 3D Shape Representation Towards Open-World Understanding”
https://colin97.github.io/OpenShape/
Apache License 2.0
236 stars 16 forks source link

Few-shot linear probing implementation ? #17

Closed alanzhangcs closed 9 months ago

alanzhangcs commented 11 months ago

Hi, thanks for your great work! I'm wondering that could you release your implementation details of your Few-shot linear probing experiment ?

kaimingkuang commented 11 months ago

Our few-shot implementation follows closely to the one in 2D CLIP. Given that features of all 3D shapes in the test dataset are extracted, we randomly choose N samples from each category in the test data and use these features to train a logistic regression model. We simply use scikit-learn's Logistic Regression with LBFGS solver, as in 2D CLIP. Next, we evaluate the trained logistic model on the rest of the test dataset. This process is repeated 10 times and we take the average performance.