Skip to content

Commit

Permalink
Merge branch 'main' of github.com:ethz-asl/radarmeetsvision into pr-i…
Browse files Browse the repository at this point in the history
…mprove-radar-pc
  • Loading branch information
marcojob committed Dec 9, 2024
2 parents 57bc989 + f306433 commit 757c645
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions scripts/validation_dataset/create_validation_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,6 @@ def generate(self):
topics = ['/image_raw', '/radar/cfar_detections', '/tf_static']
points_radar_window = []
snr_radar_window = []
noise_radar_window = []
bridge = CvBridge()
points_radar_all = None
snr_radar_all = None
Expand All @@ -106,7 +105,7 @@ def generate(self):
for i, (topic, msg, t) in enumerate(bag.read_messages(topics=topics)):
if topic == '/radar/cfar_detections':
# Transform PC to camera frame
points_radar, snr_radar, noise_radar = self.pointcloud2_to_xyz_array(msg)
points_radar, snr_radar, _ = self.pointcloud2_to_xyz_array(msg)
points_radar = points_radar.T
noise_radar = noise_radar.T

Expand Down Expand Up @@ -385,6 +384,7 @@ def process_camera_data(self, data_dict, points):


def get_depth_dict(self):
# Prepare a multiprocessing pool
pool = mp.Pool(mp.cpu_count())

# Arguments for each process
Expand Down

0 comments on commit 757c645

Please sign in to comment.