search

filaPro / oneformer3d

[CVPR2024] OneFormer3D: One Transformer for Unified Point Cloud Segmentation
Other
328 stars 33 forks source link

How to get the .pkl file? #62

Open Lizhinwafu opened 4 months ago

Lizhinwafu commented 4 months ago

image

oneformer3d-contributor commented 4 months ago

Please follow this guide.

Lizhinwafu commented 4 months ago

Thanks.

------------------ Original ------------------ From: oneformer3d-contributor @.> Date: Thu,Jun 13,2024 0:36 PM To: oneformer3d/oneformer3d @.> Cc: Lizhi Jiang @.>, Author @.> Subject: Re: [oneformer3d/oneformer3d] How to get the .pkl file? (Issue #62)

Please follow this guide.

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

Lizhinwafu commented 4 months ago

When I pre train, I also failed. image

image

It must pretrain?

oneformer3d-contributor commented 4 months ago

For pre-training you also need to download and preprocess ScanNet and Structured3D datasets. But instead of this we recommend to start from our provided checkpoint.

Lizhinwafu commented 4 months ago

Many thanks. Can you give the specific code?

image

How to change the code?

oneformer3d-contributor commented 4 months ago

simply ignore first 2 commands

Lizhinwafu commented 4 months ago

image

oneformer3d-contributor commented 4 months ago

Can you first check our checkpoint even without training from your side?

Lizhinwafu commented 4 months ago

Can you first check our checkpoint even without training from your side?

I want to train our dataset. My data only have 2 classes, which file and code line need to change?