oskarastrom commited on
Commit
24cdfbf
1 Parent(s): f4c1775

Tracker test

Browse files
Files changed (2) hide show
  1. inference.py +4 -0
  2. 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: