Beispiel #1
0
 def __getitem__(self, idx):
     print('idx:', idx)
     return _read_and_prep_v9(
         info=self._kitti_infos[idx],
         root_path=self._root_path,
         num_point_features=self._num_point_features,
         prep_func=self._prep_func)
 def __getitem__(self, idx):
     info = {
         "index": self.inform["index_list"][idx],
         "road_map": self.inform["road_map"],
         "include_roadmap": self.inform["include_roadmap"],
         "include_road_points": self.inform["include_road_points"],
         "include_roi": self.inform["include_roi"],
         "dr_area": self.inform["dr_area"]
     }
     return _read_and_prep_v9(info=info,
                              class_names=self.class_names,
                              root_path=self._root_path,
                              num_point_features=self._num_point_features,
                              argoverse_loader=self.argoverse_loader,
                              argoverse_map=self.am,
                              prep_func=self._prep_func)
Beispiel #3
0
 def __getitem__(self, idx):
     # print("[debug] image_index ", self._kitti_infos[idx]['image_idx'])
     return _read_and_prep_v9(info=self._kitti_infos[idx],
                              root_path=self._root_path,
                              num_point_features=self._num_point_features,
                              prep_func=self._prep_func)
Beispiel #4
0
 def __getitem__(self, idx):
     return _read_and_prep_v9(info=self._kitti_infos[idx],
                              root_path=self._root_path,
                              num_point_features=self._num_point_features,
                              prep_func=self._prep_func,
                              angle_deg=self._angle_deg)
Beispiel #5
0
 def __getitem__(self, idx):
     return _read_and_prep_v9(  # 读取并对点云预处理,返回example
         info=self._kitti_infos[idx],  # kitti_infos_train.pkl
         root_path=self._root_path,
         num_point_features=self._num_point_features,
         prep_func=self._prep_func)