Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,14 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: apache-2.0
|
| 3 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
---
|
| 4 |
+
# Description
|
| 5 |
+
This repository contains the fine tuned SAM2.1 model used to segment and track bees using the SAMURAI zero-shot motion tracking system. The model was fine tuned on 100 images
|
| 6 |
+
from experimental recordings of bees with manually annotated masks. The images and associated ground truth masks are available in `sam2_ft.zip`. We also provide the code used
|
| 7 |
+
to finetune the model.
|
| 8 |
+
|
| 9 |
+
The full tracking code can be found at https://github.com/AdamDHines/samurai-beetracking.
|
| 10 |
+
|
| 11 |
+
## Files
|
| 12 |
+
- `sam2.1_hiera_large.pt` -- The fine tuned SAM2.1 model.
|
| 13 |
+
- `sam2_ft.zip` -- Zip file containing raw images, segmentation masks, and bounding box annotations for fine tuning.
|
| 14 |
+
- `finetune-sam2.py` Python script containing the fine tuning code using the segmentation masks.
|