Spaces:
Runtime error
Runtime error
Commit
·
88a876b
1
Parent(s):
1548828
removed prints
Browse files
aris.py
CHANGED
@@ -341,7 +341,6 @@ def add_metadata_to_result(aris_fp, json_data, beam_width_dir=BEAM_WIDTH_DIR):
|
|
341 |
}
|
342 |
|
343 |
def get_entry(fish):
|
344 |
-
print(fish)
|
345 |
if 'marking_frame' in fish:
|
346 |
frame_num = fish['marking_frame']
|
347 |
entry = None
|
@@ -365,7 +364,6 @@ def add_metadata_to_result(aris_fp, json_data, beam_width_dir=BEAM_WIDTH_DIR):
|
|
365 |
return None # TODO better error handling
|
366 |
|
367 |
entries = []
|
368 |
-
print(json_data['fish'])
|
369 |
for fish in json_data['fish']:
|
370 |
entry = get_entry(fish)
|
371 |
entry['length'] = fish['length']*100
|
|
|
341 |
}
|
342 |
|
343 |
def get_entry(fish):
|
|
|
344 |
if 'marking_frame' in fish:
|
345 |
frame_num = fish['marking_frame']
|
346 |
entry = None
|
|
|
364 |
return None # TODO better error handling
|
365 |
|
366 |
entries = []
|
|
|
367 |
for fish in json_data['fish']:
|
368 |
entry = get_entry(fish)
|
369 |
entry['length'] = fish['length']*100
|
main.py
CHANGED
@@ -47,10 +47,8 @@ def predict_task(filepath, hyperparams, gradio_progress=None):
|
|
47 |
# run detection + tracking
|
48 |
results = do_full_inference(dataloader, image_meter_width, image_meter_height, gp=gradio_progress, hyperparams=hyperparams)
|
49 |
|
50 |
-
print(results['fish'])
|
51 |
# re-index results if desired - this should be done before writing the file
|
52 |
results = prep_for_mm(results)
|
53 |
-
print(results['fish'])
|
54 |
results = add_metadata_to_result(filepath, results)
|
55 |
results['metadata']['hyperparameters'] = hyperparams
|
56 |
|
|
|
47 |
# run detection + tracking
|
48 |
results = do_full_inference(dataloader, image_meter_width, image_meter_height, gp=gradio_progress, hyperparams=hyperparams)
|
49 |
|
|
|
50 |
# re-index results if desired - this should be done before writing the file
|
51 |
results = prep_for_mm(results)
|
|
|
52 |
results = add_metadata_to_result(filepath, results)
|
53 |
results['metadata']['hyperparameters'] = hyperparams
|
54 |
|