Upload README.md

README.md

@@ -1,12 +1,29 @@
----
-title: Pokemon
-emoji: 📈
-colorFrom: green
-colorTo: purple
-sdk: gradio
-sdk_version: 5.49.1
-app_file: app.py
-pinned: false
----
+# Pokemon object detections
+Using yolov8 after training with Google Colab
+## Dataset
+Because of lack of data, there are only 7 classes: ```pikachu```, ```charmander```, ```bulbasaur```, ```squirtle```, ```eevee```, ```jigglypuff``` and ```other```.  
+  
+![labels](https://github.com/vovod/yolov8-pokemon-object-detection/blob/main/after_train/labels.jfif)
+## Requirements
+```
+pip install ultralytics
+```
+## Preprocess Data
+The ```convert.py``` used to convert *.xml* label file to *.txt* yolo label file.  
+Run ```resize_image.py``` to resize image's width to 640.
+## Train with Colab
+Edit ```name.yaml```.
+Upload images and labels.
+```
+!yolo train model=yolov8n.pt data=/content/name.yaml epochs=50 imgsz=640
+```
+## Training's Result
+```last.pt``` and ```best.pt``` in result folder.
+  
+![train](https://github.com/vovod/yolov8-pokemon-object-detection/blob/main/after_train/results.png?raw=true)
+## Predict
+Run ```predict.py``` to see result. This is my predict to ```test.mp4```:  
+  
+![result](https://github.com/vovod/yolov8-pokemon-object-detection/blob/main/result.gif?raw=true)
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+#### Thank you for stopping by!