Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
Former-commit-id: a237248
  • Loading branch information
PetervanLunteren authored Nov 16, 2022
1 parent 21df068 commit 890eced
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -100,12 +100,6 @@ EcoAssist will automatically run on Windows and Linux if compatible `CUDA` GPU i
## Custom model support
EcoAssist can run custom `yolov5` models if they are retrained from the MegaDetector model using transfer learning. For example, if you find that MegaDetector is not great at recognising a certain species as "animal", you can retrain the model and add some labelled data of the cases you want it to improve on. You can also expand on the three default classes ("animal", "person" and "vehicle") and train the model to detect custom classes (e.g. "species A" and "species B"). If you add classes, you’ll need to adjust `run_detector.py` and `separate_detections_into_folders.py` too. If you need any help with that, [let me know](mailto:[email protected])!

So basically, EcoAssist can run your custom model if it works with the same command as MegaDetector.

```bash
python run_detector_batch.py "your_custom_model.pt" "/some/image/folder" "output.json"
```

## Citation
If you use EcoAssist in your research, don't forget to cite the engine behind EcoAssist ([MegaDetector](https://github.com/microsoft/CameraTraps/blob/main/megadetector.md)) and the EcoAssist software itself.
```BibTeX
Expand Down

0 comments on commit 890eced

Please sign in to comment.