jonathan-roos commited on
Commit
f70cb67
1 Parent(s): c6f57a1

changed model to densenet

Browse files
Files changed (1) hide show
  1. detector.py +1 -1
detector.py CHANGED
@@ -56,7 +56,7 @@ def crop_bat(img, box):
56
  def find_bats(allImgs):
57
  total_bats, batDepthMin,batDepthMax = 0, 50, 400
58
  allImgAfter = []
59
- learn = load_learner("model.pkl")
60
 
61
  for img in allImgs:
62
  blobs = cv.findContours(img[1], cv.RETR_LIST, cv.CHAIN_APPROX_SIMPLE)[-2]
 
56
  def find_bats(allImgs):
57
  total_bats, batDepthMin,batDepthMax = 0, 50, 400
58
  allImgAfter = []
59
+ learn = load_learner("model_densenet_zeros.pkl")
60
 
61
  for img in allImgs:
62
  blobs = cv.findContours(img[1], cv.RETR_LIST, cv.CHAIN_APPROX_SIMPLE)[-2]