直接上代码。 defload_pointcloud(pts_filename):"""读取点云文件返回 np.array, shape(N, 3)"""# 加载点云mmcv.check_file_exist(pts_filename)ifpts_filename.endswith('.npy'):points=np.load(pts_filename)else:points=np.fromfile(pts_filename,dtype=np.float32)# 转换点云格式points=points.re...
+"""Script to gather benchmarked models and prepare them for upload. + +Usage: +python gather_models.py ${root_path} ${out_dir} +""" + +import argparse +import glob +import json +import mmcv +import shutil +import subprocess +import torch +from os import path as osp + +# build ...
if point_sample is None: raise RuntimeError('Please install mmcv-full for ' 'point_sample ops') self.num_fcs = num_fcs self.coarse_pred_each_layer = coarse_pred_each_layer fc_in_channels = sum(self.in_channels) + self.num_classes fc_channels = self.channels self.fcs = ...
(self,cloud_file,i):# load sample from file#self.points = self.load_cloud_from_nuscenes_file(cloud_file)#self.points = self.load_cloud_from_deecamp_file(cloud_file)self.points=self.load_cloud_from_my_file(cloud_file)# prepare inputvoxels,coords,num_points=self.voxel_generator.generate(...
InstallThe project has been fully tested under MMDetection V2.22.0 and MMCV V1.4.6, other versions may not be compatible. so you have to install mmcv and mmdetection firstly. You can refer to Installation of MMCV & Installation of MMDetection...
(data_loaders, cfg.workflow, cfg.total_epochs) File "/opt/conda/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 125, in run epoch_runner(data_loaders[i], **kwargs) File "/opt/conda/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 47, in ...
File "/home/jidong/anaconda3/envs/mmrazor/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 51, in train self.run_iter(data_batch, train_mode=True, **kwargs) File "/home/jidong/anaconda3/envs/mmrazor/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py",...
Acknowledgement The code is devloped based on mmdetection, some part of codes are borrowed from SA-SSD, SECOND, and PointRCNN. mmdetection mmcv second.pytorch PointRCNN SA-SSD Contact If you have questions, you can contact qianruii@126.com.About...
The code is devloped based on mmdetection, some part of codes are borrowed from SECOND and PointRCNN. mmdetection mmcv second.pytorch PointRCNN Releases No releases published Packages No packages published Languages Python93.7% Cuda3.4% C++2.9%...