Hugging Face's logo

Spaces:
NN-BRD
/
hackathon_depth_segment
Runtime error

App Files Community
jens commited on
Commit
55bcfca
·
1 Parent(s): 15a3702

fix

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -31,7 +31,7 @@ def PCL(mask, depth):
31
  print(np.unique(colors, axis=0))
32
  print(np.unique(colors, axis=1))
33
  print(np.unique(colors))
34
- mask = (colors[:, 0] == 255)
35
  print(mask.sum())
36
  print(colors.shape)
37
  points = points[mask]
 
31
  print(np.unique(colors, axis=0))
32
  print(np.unique(colors, axis=1))
33
  print(np.unique(colors))
34
+ mask = (colors[:, 0] == 1.)
35
  print(mask.sum())
36
  print(colors.shape)
37
  points = points[mask]