Skip to content

UniSeg3D model

Latest
Compare
Choose a tag to compare
@chunshengshi chunshengshi released this 19 Jul 06:43
· 11 commits to main since this release

model_best.pth: The trained model without further finetuning.
model_best.log: The training log of the model_best.pth.
model_best_2.pth: Fine-tuning the model_best.pth with smaller learning rate and weight decay.
uniseg3d_infos_val.pkl: Processed val-set data infos.
uniseg3d_infos_train.pkl: Processed train-set data infos.
scannet_cls_embedding.pth: Processed class embeddings.
sstnet_scannet.pth: Pretrained SSTNet used to initialize sparse 3D U-Net backbone.