search

guochengqian / PointNeXt

[NeurIPS'22] PointNeXt: Revisiting PointNet++ with Improved Training and Scaling Strategies
https://guochengqian.github.io/PointNeXt/
MIT License
770 stars 111 forks source link

What is num_per_class variable s3dis.py #129

Closed Husnain-Javid closed 1 year ago

Husnain-Javid commented 1 year ago

Hi @guochengqian, thanks for your work. I have a quick question.

What is num_per_class variable in openpoints/dataset/s3dis/s3dis.py

num_per_class = np.array([3370714, 2856755, 4919229, 318158, 375640, 478001, 974733, 650464, 791496, 88727, 1284130, 229758, 2272837], dtype=np.int32)

I am trying to train PointNext on the custom dataset. I have prepared my dataset according to your provided S3DIS dataset. Which has 

- data
        - S3DIS
                  - s3disfull
                                - raw
                                        - .....npy
                                - processed
                                        - .....pkl

Do I need to use s3dis.py in openponits?

guochengqian commented 1 year ago

It is the number of points per class (class mean the ground truth annonation: door, chair, xxx). It is only used in weighted loss function. You can follow s3dis.py to create your data loader.