Denys Rozumnyi
commited on
Commit
·
b02e5d5
1
Parent(s):
87a5a37
update
Browse files- geom_solver.py +3 -1
- 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 : [
|
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
|
|