KPConv-PyTorch icon indicating copy to clipboard operation
KPConv-PyTorch copied to clipboard

How to decide the "in_feature_dim" of train_**.py?

Open GeoSur opened this issue 10 months ago • 1 comments

if self.config.in_features_dim == 1: pass elif self.config.in_features_dim == 4: stacked_features = np.hstack((stacked_features, features[:, :3])) elif self.config.in_features_dim == 5: stacked_features = np.hstack((stacked_features, features)) elif self.config.in_features_dim == 7: stacked_features = np.hstack((stacked_features, stacked_points, features[:, :3]))

As this section described, we could select 1,4,5 and 7 as the feature dim, but 1 and 4 represents without and with RGB respectively, but how to decide 5 and 7?

GeoSur avatar Apr 02 '24 12:04 GeoSur

You can define the in_features_dim in the config class, for example S3DIS uses 5 features and it's defined on line 146 :

https://github.com/HuguesTHOMAS/KPConv-PyTorch/blob/435d117c1bb2f82d7030a669444b8f20976b7a6a/train_S3DIS.py#L144-L146

LucaRom avatar May 09 '24 18:05 LucaRom