Skip to content

Latest commit

 

History

History
23 lines (14 loc) · 491 Bytes

README.md

File metadata and controls

23 lines (14 loc) · 491 Bytes
Raw

WildAnimalDetection

The code implements a trap-camera image classification system for Jasper Ridge (JR) Biological Preserve. best_model.pt is needed but not included in the github due to its large size.

Author: Yifan Yu(@YifanYu5124), Yancheng Li(@lycyeah) and Tianpei Qian(@qiantianpei)

Dependencies

  • Python 3.6
  • PyTorch 0.3

Files

  • predict.py: main file for prediction
  • WildAnimalDataset.py: dataloader

Prediction

python predict.py --threshold 0.9