From 890eceded997ec992d825fb5a82b3af64830490c Mon Sep 17 00:00:00 2001 From: Peter van Lunteren <85185478+PetervanLunteren@users.noreply.github.com> Date: Thu, 17 Nov 2022 07:35:42 +1100 Subject: [PATCH] Update README.md Former-commit-id: a237248b617c25cdabbb5d518543b520564a8468 --- README.md | 6 ------ 1 file changed, 6 deletions(-) diff --git a/README.md b/README.md index 7a32775d..ee60d968 100644 --- a/README.md +++ b/README.md @@ -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:petervanlunteren@hotmail.com)! -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