fix: spconv error

HuangCongQing · Dec 4, 2022 · cce3e23 · cce3e23
1 parent 65199c8
commit cce3e23
Show file tree

Hide file tree

Showing 15 changed files with 839 additions and 72 deletions.
diff --git a/.gitignore b/.gitignore
@@ -7,4 +7,5 @@ painted_lidar
 **/.DS_Store
 **/__pycache__
 # 数据
-detector/data
+detector/data
+detector/output
diff --git a/.idea/.gitignore b/.idea/.gitignore
diff --git a/.idea/PointPainting.iml b/.idea/PointPainting.iml
diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml
diff --git a/.idea/inspectionProfiles/profiles_settings.xml b/.idea/inspectionProfiles/profiles_settings.xml
diff --git a/.idea/misc.xml b/.idea/misc.xml
diff --git a/.idea/modules.xml b/.idea/modules.xml
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
diff --git a/detector/pcdet/datasets/dataset.py b/detector/pcdet/datasets/dataset.py
@@ -30,8 +30,14 @@ def __init__(self, dataset_cfg=None, class_names=None, training=True, root_path=
         self.data_augmentor = DataAugmentor(
             self.root_path, self.dataset_cfg.DATA_AUGMENTOR, self.class_names, logger=self.logger
         ) if self.training else None
+        # # 定义数据处理类
+        # 解决：_init__() missing 1 required positional argument: 'num_point_features
+        # self.data_processor = DataProcessor(
+        #     self.dataset_cfg.DATA_PROCESSOR, point_cloud_range=self.point_cloud_range, training=self.training
+        # )
         self.data_processor = DataProcessor(
-            self.dataset_cfg.DATA_PROCESSOR, point_cloud_range=self.point_cloud_range, training=self.training
+            self.dataset_cfg.DATA_PROCESSOR, point_cloud_range=self.point_cloud_range,
+            training=self.training, num_point_features=self.point_feature_encoder.num_point_features
         )
 
         self.grid_size = self.data_processor.grid_size