diff --git a/src/datamodules/components/kitti_dataset.py b/src/datamodules/components/kitti_dataset.py index b1e8be7..d11e57f 100644 --- a/src/datamodules/components/kitti_dataset.py +++ b/src/datamodules/components/kitti_dataset.py @@ -380,8 +380,7 @@ def format_label(self, line): # Locattion: x, y, z Location = [line[11], line[12], line[13]] - # bring the KITTI center up to the middle of the object - Location[1] -= Dimension[0] / 2 + # KITTI Location == center of 3D Object Orientation = np.zeros((self.bins, 2)) Confidence = np.zeros(self.bins) @@ -753,4 +752,4 @@ def get_augmentation(self, data): results2 = (time() - start2) * 1000 # print("KITTI Dataset: {} ms".format(results1)) - print("KITTI Dataset3: {} ms".format(results2)) \ No newline at end of file + print("KITTI Dataset3: {} ms".format(results2)) diff --git a/src/utils/Plotting.py b/src/utils/Plotting.py index 6ea901b..39b985e 100644 --- a/src/utils/Plotting.py +++ b/src/utils/Plotting.py @@ -366,7 +366,7 @@ def compute_3Dbox(self, P2, line): z_corners = [0, 0, 0, obj.w, obj.w, obj.w, obj.w, 0] # -w/2 x_corners = [i - obj.l / 2 for i in x_corners] - y_corners = [i - obj.h for i in y_corners] + y_corners = [i - obj.h / 2 for i in y_corners] z_corners = [i - obj.w / 2 for i in z_corners] corners_3D = np.array([x_corners, y_corners, z_corners]) @@ -683,4 +683,4 @@ def save_plot(self, path, name): mode="training", ) - plot.visualization() \ No newline at end of file + plot.visualization()