Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+results.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,45 @@
+---
+library_name: ultralytics
+tags:
+  - object-detection
+  - yolo
+  - yolo9
+  - animals
+  - CAN-Benchmark
+license: mit
+datasets:
+  - ICICLE-AI/CAN_Benchmark
+task: object-detection
+---
+
+# YOLOv9 – Animal Detection (Zebra, Impala, Giraffe)
+
+This model is a **YOLOv9** detector fine-tuned with [Ultralytics](https://github.com/ultralytics/ultralytics).  
+It was trained for **50 epochs** on a **subset** of the [ICICLE-AI/CAN_Benchmark](https://huggingface.co/datasets/ICICLE-AI/CAN_Benchmark) dataset containing three species:  
+
+- **0 → zebra**  
+- **1 → impala**  
+- **2 → giraffe**  
+
+## 📊 Training details
+- Framework: Ultralytics YOLOv9  
+- Epochs: 50  
+- See full hyperparameters in [`args.yaml`](./args.yaml)
+
+The model converged by ~40 epochs, and shows strong precision/recall on the held-out validation set:
+
+![Training curves](./results.png)
+
+## 🚀 Usage
+
+Load the model directly from the Hugging Face Hub:
+
+```python
+from ultralytics import YOLO
+
+# Load model from HF Hub
+model = YOLO("ICICLE-AI/yolov9-animals-AE-data")
+
+# Run inference
+results = model("demo.jpg")
+results[0].show()

args.yaml

@@ -0,0 +1,106 @@
+task: detect
+mode: train
+model: yolov9c.pt
+data: yolo_data.yaml
+epochs: 50
+time: null
+patience: 100
+batch: 16
+imgsz: 640
+save: true
+save_period: -1
+cache: false
+device: 0,1,2,3
+workers: 8
+project: null
+name: train
+exist_ok: false
+pretrained: true
+optimizer: auto
+verbose: true
+seed: 0
+deterministic: true
+single_cls: false
+rect: false
+cos_lr: false
+close_mosaic: 10
+resume: false
+amp: true
+fraction: 1.0
+profile: false
+freeze: null
+multi_scale: false
+compile: false
+overlap_mask: true
+mask_ratio: 4
+dropout: 0.0
+val: true
+split: val
+save_json: false
+conf: null
+iou: 0.7
+max_det: 300
+half: false
+dnn: false
+plots: true
+source: null
+vid_stride: 1
+stream_buffer: false
+visualize: false
+augment: false
+agnostic_nms: false
+classes: null
+retina_masks: false
+embed: null
+show: false
+save_frames: false
+save_txt: false
+save_conf: false
+save_crop: false
+show_labels: true
+show_conf: true
+show_boxes: true
+line_width: null
+format: torchscript
+keras: false
+optimize: false
+int8: false
+dynamic: false
+simplify: true
+opset: null
+workspace: null
+nms: false
+lr0: 0.01
+lrf: 0.01
+momentum: 0.937
+weight_decay: 0.0005
+warmup_epochs: 3.0
+warmup_momentum: 0.8
+warmup_bias_lr: 0.1
+box: 7.5
+cls: 0.5
+dfl: 1.5
+pose: 12.0
+kobj: 1.0
+nbs: 64
+hsv_h: 0.015
+hsv_s: 0.7
+hsv_v: 0.4
+degrees: 0.0
+translate: 0.1
+scale: 0.5
+shear: 0.0
+perspective: 0.0
+flipud: 0.0
+fliplr: 0.5
+bgr: 0.0
+mosaic: 1.0
+mixup: 0.0
+cutmix: 0.0
+copy_paste: 0.0
+copy_paste_mode: flip
+auto_augment: randaugment
+erasing: 0.4
+cfg: null
+tracker: botsort.yaml
+save_dir: /users/PAS2271/thevyasamit/Desktop/Projects/fine_tune_yolo9/runs/detect/train

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bddf4ba5e45edcfceaf9f5d28c7b0ef683555af141d6652414dbf4986d8f57bb
+size 51569548

yolo_data.yaml

@@ -0,0 +1,6 @@
+path: .
+train: autosplit_train.txt
+val: autosplit_val.txt
+test: autosplit_test.txt
+nc: 3
+names: ['zebra','impala','giraffe']