Denys Rozumnyi commited on
Commit
b02e5d5
·
1 Parent(s): 87a5a37
Files changed (2) hide show
  1. geom_solver.py +3 -1
  2. testing.ipynb +0 -0
geom_solver.py CHANGED
@@ -178,7 +178,7 @@ class GeomSolver(object):
178
  uvs.append(uv)
179
 
180
  edges = []
181
- thresholds_min_mean = {0 : [5, 7], 1 : [15, 25], 2: [30, 1000]}
182
  for i in range(pyt_centers.shape[0]):
183
  for j in range(i+1, pyt_centers.shape[0]):
184
  etype = (self.is_apex[i] + self.is_apex[j])
@@ -193,6 +193,8 @@ class GeomSolver(object):
193
 
194
  if not ( (cki in center_visibility[i]) or (cki in center_visibility[j]) ):
195
  continue
 
 
196
 
197
  uvi = torch.round(self.pyt_cameras[ki].transform_points(points_inter)[:, :2]).cpu().numpy().astype(int)
198
  if (uvi <= 0).any() or (uvi[:,0] >= self.width).any() or (uvi[:,1] >= self.height).any():
 
178
  uvs.append(uv)
179
 
180
  edges = []
181
+ thresholds_min_mean = {0 : [5, 7], 1 : [9, 25], 2: [30, 1000]}
182
  for i in range(pyt_centers.shape[0]):
183
  for j in range(i+1, pyt_centers.shape[0]):
184
  etype = (self.is_apex[i] + self.is_apex[j])
 
193
 
194
  if not ( (cki in center_visibility[i]) or (cki in center_visibility[j]) ):
195
  continue
196
+ if self.broken_cams[ki]:
197
+ continue
198
 
199
  uvi = torch.round(self.pyt_cameras[ki].transform_points(points_inter)[:, :2]).cpu().numpy().astype(int)
200
  if (uvi <= 0).any() or (uvi[:,0] >= self.width).any() or (uvi[:,1] >= self.height).any():
testing.ipynb CHANGED
The diff for this file is too large to render. See raw diff