SimingYan / IAE

[ICCV 2023] "Implicit Autoencoder for Point-Cloud Self-Supervised Representation Learning"
98 stars 17 forks source link

Linear evaluation on ModelNet40 #16

Closed Kracozebr closed 1 year ago

Kracozebr commented 1 year ago

Hi, first of all thanks for sharing your code.

I'm trying to reproduce Linear evaluation on ModelNet40

I downloaded the dataset and run the evaluation:

python train_svm.py --encoder=dgcnn_cls --restore_path=./pretrained_models/modelnet40_svm.pt

And get the next result:

Transfer linear SVM accuracy: 92.06%

How to achieve the declared 94.2?

SimingYan commented 1 year ago

Hi,

The result of linear evaluation is 92.06%, which you have already achieved. And 94.2 is the fine-tuned classification result. This is different from linear evaluation.

Kracozebr commented 1 year ago

Ok, thanks for clarification.