KPConv-PyTorch
KPConv-PyTorch copied to clipboard
How to decide the "in_feature_dim" of train_**.py?
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?
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