Commit d86dffa4 authored by liyinhao's avatar liyinhao
Browse files

change mean_color to color_mean

parent 9afa546a
...@@ -15,13 +15,13 @@ class LoadPointsFromFile(object): ...@@ -15,13 +15,13 @@ class LoadPointsFromFile(object):
name (str): scannet or sunrgbd. name (str): scannet or sunrgbd.
use_color (bool): Whether to use color. use_color (bool): Whether to use color.
use_height (bool): Whether to use height. use_height (bool): Whether to use height.
mean_color (List[float]): Mean color of the point cloud. color_mean (List[float]): Mean color of the point cloud.
""" """
def __init__(self, use_color, use_height, mean_color): def __init__(self, use_color, use_height, color_mean):
self.use_color = use_color self.use_color = use_color
self.use_height = use_height self.use_height = use_height
self.mean_color = mean_color self.color_mean = color_mean
def __call__(self, results): def __call__(self, results):
data_path = results.get('data_path', None) data_path = results.get('data_path', None)
...@@ -44,7 +44,7 @@ class LoadPointsFromFile(object): ...@@ -44,7 +44,7 @@ class LoadPointsFromFile(object):
pcl_color = point_cloud[:, 3:6] pcl_color = point_cloud[:, 3:6]
point_cloud = point_cloud[:, 0:6] point_cloud = point_cloud[:, 0:6]
point_cloud[:, 3:] = (point_cloud[:, 3:] - point_cloud[:, 3:] = (point_cloud[:, 3:] -
np.array(self.mean_color)) / 256.0 np.array(self.color_mean)) / 256.0
if self.use_height: if self.use_height:
floor_height = np.percentile(point_cloud[:, 2], 0.99) floor_height = np.percentile(point_cloud[:, 2], 0.99)
...@@ -60,7 +60,7 @@ class LoadPointsFromFile(object): ...@@ -60,7 +60,7 @@ class LoadPointsFromFile(object):
repr_str = self.__class__.__name__ repr_str = self.__class__.__name__
repr_str += '(use_height={})'.format(self.use_height) repr_str += '(use_height={})'.format(self.use_height)
repr_str += '(use_color={}'.format(self.use_color) repr_str += '(use_color={}'.format(self.use_color)
repr_str += '(mean_color={})'.format(self.mean_color) repr_str += '(mean_color={})'.format(self.color_mean)
return repr_str return repr_str
def _get_lidar(self, scan_name, data_path): def _get_lidar(self, scan_name, data_path):
......
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