abie-e / BFTT3D

5 stars 0 forks source link

BFTT3D

This repository is for implementing frameworks introduced in the following paper:
"Backpropagation-free Network for 3D Test-time Adaptation"

Installation

Create a conda environment and install dependencies:

conda env create -f environment.yml

If this fails, please try to install the Point-NN and MATE environments one by one. 1.Point-NN 2.MATE

Dataset

Folder structure

.
├── ...
├── datasets
├── data                    
│   ├── modelnet40_ply_hdf5_2048          
│   ├── modelnet40_c         
│   └── ...                
├── checkpoint                   
│   └── ... 
└── ....

Please download the following datasets for the prototype memory

Download ModelNet40

Download ScanObjectNN

Download ShapeNetCoreV2

Corruption:

Please follow the script from ModelNet40-C to add corruption.

We also provide download links:

Download ModelNet40-C from Google Drive.

Download ScanObjectNN-C from Google Drive.

Checkpoint:

checkpoint is uploaded here.

BFTT3D

For ModelNet40 dataset, just run:

python run_BFTT3D_mn40.py --pth pointnet

The code is not fully organized and may not look polished. Refactoring and finalization are underway.