Skip to content

GitLab

Commit 329748c0 authored by liyinhao's avatar liyinhao
Browse files

add assertion

parent 470a58ad
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
mmdet3d/datasets/pipelines/indoor_augment.py
View file @ 329748c0
...@@ -147,6 +147,7 @@ class IndoorGlobalRotScale(object): ...@@ -147,6 +147,7 @@ class IndoorGlobalRotScale(object):
# Augment RGB color # Augment RGB color
if self.use_color: if self.use_color:
assert points.shape[1] >= 6
rgb_color = points[:, 3:6] + self.color_mean rgb_color = points[:, 3:6] + self.color_mean
# brightness change for each channel # brightness change for each channel
rgb_color *= (1 + 0.4 * np.random.random(3) - 0.2) rgb_color *= (1 + 0.4 * np.random.random(3) - 0.2)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment