Releases: dk-liang/UniSeg3D
Releases · dk-liang/UniSeg3D
UniSeg3D model
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.