Initial Commit

.gitignore

@@ -0,0 +1,2 @@
+*.jpg
+*.csv

README.md

@@ -1,3 +1,47 @@
 ---
 license: apache-2.0
+library: ultralytics
+tags:
+- object-detection
+- pytorch
+- roboflow-universe
+- pickle
+- face-detection
 ---
+# Face Detection using YOLOv8
+
+This model was fine tuned on a dataset of over 10k images containing human faces. The model was fine tuned for 100 epochs with a batch size of 16 on a single NVIDIA V100 16GB GPU, it took around 140 minutes for the fine tuning to complete.
+
+## Downstream Tasks
+
+- __Face Detection__: This model can directly use this model for face detection or it can be further fine tuned own a custom dataset to improve the prediction capabilities.
+- __Face Recognition__: This model can be fine tuned to for face recognition tasks as well, create a dataset with the images of faces and label them accordingly using name or any ID and then use this model as a base model for fine tuning.
+
+# Example Usage
+
+```python
+# load libraries
+from huggingface_hub import hf_hub_download
+from ultralytics import YOLO
+from supervision import Detections
+from PIL import Image
+
+# download model
+model_path = hf_hub_download(repo_id="arnabdhar/YOLOv8-Face-Detection", filename="model.pt")
+
+# load model
+model = YOLO(model_path)
+
+# inference
+image_path = "/path/to/image"
+output = YOLO(Image.open("image_path"))
+results = Detections.from_ultralytics(output[0])
+
+```
+
+
+# Links
+
+- __Dataset Source__: [Roboflow Universe](https://universe.roboflow.com)
+- __Weights & Biases__: [Run Details](https://wandb.ai/2wb2ndur/Face-Detection/overview?workspace=user-2wb2ndur) 
+- __Training Artifacts__: [training-artifacts](./fine-tune-artifacts/)

events.out.tfevents.1699172798.7b5d18c3a8a9.250.0

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

fine-tune-artifacts/args.yaml

@@ -0,0 +1,98 @@
+task: detect
+mode: train
+model: yolov8n.pt
+data: data.yaml
+epochs: 20
+patience: 50
+batch: 16
+imgsz: 640
+save: true
+save_period: -1
+cache: disk
+device: 0
+workers: 8
+project: Face-Detection
+name: laced-silence-2
+exist_ok: false
+pretrained: true
+optimizer: AdamW
+verbose: true
+seed: 0
+deterministic: true
+single_cls: false
+rect: false
+cos_lr: true
+close_mosaic: 10
+resume: false
+amp: true
+fraction: 1.0
+profile: false
+freeze: null
+overlap_mask: true
+mask_ratio: 4
+dropout: 0.0
+val: true
+split: val
+save_json: false
+save_hybrid: false
+conf: null
+iou: 0.7
+max_det: 300
+half: false
+dnn: false
+plots: true
+source: null
+show: false
+save_txt: false
+save_conf: false
+save_crop: false
+show_labels: true
+show_conf: true
+vid_stride: 1
+stream_buffer: false
+line_width: null
+visualize: false
+augment: false
+agnostic_nms: false
+classes: null
+retina_masks: false
+boxes: true
+format: torchscript
+keras: false
+optimize: false
+int8: false
+dynamic: false
+simplify: false
+opset: null
+workspace: 4
+nms: false
+lr0: 3.0e-05
+lrf: 0.01
+momentum: 0.937
+weight_decay: 0.0005
+warmup_epochs: 5
+warmup_momentum: 0.8
+warmup_bias_lr: 0.1
+box: 7.5
+cls: 0.5
+dfl: 1.5
+pose: 12.0
+kobj: 1.0
+label_smoothing: 0.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
+mosaic: 1.0
+mixup: 0.0
+copy_paste: 0.0
+cfg: null
+tracker: botsort.yaml
+save_dir: Face-Detection/laced-silence-2

model.pt

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