Spaces:
Runtime error
Runtime error
Commit
•
24cdfbf
1
Parent(s):
f4c1775
Tracker test
Browse files- inference.py +4 -0
- lib/fish_eye/tracker.py +1 -1
inference.py
CHANGED
@@ -154,6 +154,10 @@ def do_detection(dataloader, model, device, gp=None, batch_size=BATCH_SIZE, verb
|
|
154 |
with torch.no_grad():
|
155 |
inf_out, _ = model(img, augment=False)
|
156 |
|
|
|
|
|
|
|
|
|
157 |
# Save shapes for resizing to original shape
|
158 |
batch_shape = []
|
159 |
for si, pred in enumerate(inf_out):
|
|
|
154 |
with torch.no_grad():
|
155 |
inf_out, _ = model(img, augment=False)
|
156 |
|
157 |
+
print(inf_out.shape)
|
158 |
+
print(inf_out[1, :])
|
159 |
+
print(inf_out[:, 1])
|
160 |
+
|
161 |
# Save shapes for resizing to original shape
|
162 |
batch_shape = []
|
163 |
for si, pred in enumerate(inf_out):
|
lib/fish_eye/tracker.py
CHANGED
@@ -131,7 +131,7 @@ class Tracker:
|
|
131 |
json_data['fish'].append(fish_entry)
|
132 |
|
133 |
|
134 |
-
# filter 'fish' field by fish length
|
135 |
json_data = Fish_Length.add_lengths(json_data)
|
136 |
invalid_ids = []
|
137 |
if min_length != -1.0:
|
|
|
131 |
json_data['fish'].append(fish_entry)
|
132 |
|
133 |
|
134 |
+
# filter 'fish' field by fish length and travel distance
|
135 |
json_data = Fish_Length.add_lengths(json_data)
|
136 |
invalid_ids = []
|
137 |
if min_length != -1.0:
|