koolguy06 commited on
Commit
6c8fe04
·
verified ·
1 Parent(s): c7fb242

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -1,5 +1,7 @@
1
  ---
2
  license: mit
 
 
3
  ---
4
  ## Model Training
5
 
@@ -26,4 +28,4 @@ The training was performed using the Ultralytics YOLOv8 framework. The process i
26
  1. **Dataset Preparation**: Downloaded the Augmented Startups Playing Cards dataset from Roboflow in YOLOv8 format
27
  2. **Model Initialization**: Started with pre-trained YOLOv8l weights (`yolov8l.pt`)
28
  3. **Fine-tuning**: Trained for 50 epochs on the playing cards dataset
29
- 4. **Model Export**: Saved the fine-tuned model as `playing-cards.pt`
 
1
  ---
2
  license: mit
3
+ base_model:
4
+ - Ultralytics/YOLOv8l
5
  ---
6
  ## Model Training
7
 
 
28
  1. **Dataset Preparation**: Downloaded the Augmented Startups Playing Cards dataset from Roboflow in YOLOv8 format
29
  2. **Model Initialization**: Started with pre-trained YOLOv8l weights (`yolov8l.pt`)
30
  3. **Fine-tuning**: Trained for 50 epochs on the playing cards dataset
31
+ 4. **Model Export**: Saved the fine-tuned model as `playing-cards.pt`