nd pred_scores shape must be conf.batch_size x conf.max_n_animals x conf.n_classes x 2 and conf.batch_size x conf.max_n_animals x conf.n_classes
              
              <b>return {'locs':pred_locs, 'conf': pred_scores}</b>
              
          <b>def close_fn():</b>
              # Close any open files, clear GPU memory etc. 
              torch.cuda.empty_cache()
        
          <b>return pred_fn, close_fn, model_file_used</b>
      
</code></pre>

<p>
<h2><a id="conf">Configuration Object</a><