oskarastrom commited on
Commit
88a876b
·
1 Parent(s): 1548828

removed prints

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