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
244 stars 17 forks source link

Few-shot linear probing implementation ? #17

Closed alanzhangcs closed 10 months ago

alanzhangcs commented 1 year 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 1 year 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.