Skip to content

MegaDetector notebook: download dataset and fine-tune#71

Merged
nee1k merged 1 commit intomainfrom
feature/megadetector-notebook-download-finetune
Feb 3, 2026
Merged

MegaDetector notebook: download dataset and fine-tune#71
nee1k merged 1 commit intomainfrom
feature/megadetector-notebook-download-finetune

Conversation

@nee1k
Copy link
Copy Markdown
Collaborator

@nee1k nee1k commented Feb 3, 2026

Summary

  • Notebook: Add optional section 3b to download more images (COCO128), remap classes to MegaDetector (animal, person, vehicle), and fine-tune via YOLOv5 training.
  • .gitignore: Ignore examples/notebooks/data/, runs/, and yolov5/ so downloaded data and cloned repo are not committed.

Changes

  • Download COCO128, map to 3 classes, write data/finetune_md/data.yaml
  • Run YOLOv5 train.py from MegaDetector weights (clone repo if needed)
  • Optional cell to load fine-tuned best.pt for inference
  • .gitignore updated for notebook data/runs/yolov5

…nore

- Add section 3b: download COCO128, remap to animal/person/vehicle, prepare data.yaml
- Add YOLOv5 training from MegaDetector weights (clone repo + train.py)
- Add optional cell to load fine-tuned best.pt for inference
- Ignore examples/notebooks/data/, runs/, yolov5/
@nee1k nee1k merged commit 465f5d2 into main Feb 3, 2026
1 check passed
@nee1k nee1k deleted the feature/megadetector-notebook-download-finetune branch February 3, 2026 22:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant