Skip to content

Commit

Permalink
Iterate over the list to get all values. No need to create an np arra…
Browse files Browse the repository at this point in the history
…y since we are only reading.
  • Loading branch information
romulogoncalves committed Apr 11, 2018
1 parent 2c6d487 commit 5f5990b
Showing 1 changed file with 1 addition and 2 deletions.
Expand Up @@ -52,8 +52,7 @@ def extract(self, point_cloud, neighborhood, target_point_cloud, target_index, v
:param volume_description: volume object that describes the shape and size of the search volume
:return: feature value
"""
class_neighbors = np.array(point_cloud[point]['raw_classification']["data"][
neighborhood])
class_neighbors = [point_cloud[point]['raw_classification']["data"][n] for n in neighborhood]

ground_indices = self._get_ground_indices(
class_neighbors, self.ground_tags)
Expand Down

0 comments on commit 5f5990b

Please sign in to comment.