Dirake commited on
Commit
fd89247
·
verified ·
1 Parent(s): 90a4982

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -30
README.md CHANGED
@@ -1,30 +1,8 @@
1
-
2
- # Pokemon object detections
3
- Using yolov8 after training with Google Colab
4
- ## Dataset
5
- Because of lack of data, there are only 7 classes: ```pikachu```, ```charmander```, ```bulbasaur```, ```squirtle```, ```eevee```, ```jigglypuff``` and ```other```.
6
-
7
- ![labels](https://github.com/vovod/yolov8-pokemon-object-detection/blob/main/after_train/labels.jfif)
8
- ## Requirements
9
- ```
10
- pip install ultralytics
11
- ```
12
- ## Preprocess Data
13
- The ```convert.py``` used to convert *.xml* label file to *.txt* yolo label file.
14
- Run ```resize_image.py``` to resize image's width to 640.
15
- ## Train with Colab
16
- Edit ```name.yaml```.
17
- Upload images and labels.
18
- ```
19
- !yolo train model=yolov8n.pt data=/content/name.yaml epochs=50 imgsz=640
20
- ```
21
- ## Training's Result
22
- ```last.pt``` and ```best.pt``` in result folder.
23
-
24
- ![train](https://github.com/vovod/yolov8-pokemon-object-detection/blob/main/after_train/results.png?raw=true)
25
- ## Predict
26
- Run ```predict.py``` to see result. This is my predict to ```test.mp4```:
27
-
28
- ![result](https://github.com/vovod/yolov8-pokemon-object-detection/blob/main/result.gif?raw=true)
29
-
30
- #### Thank you for stopping by!
 
1
+ title: "Pokemon Object Detection"
2
+ emoji: 🐉
3
+ colorFrom: indigo
4
+ colorTo: pink
5
+ sdk: gradio
6
+ sdk_version: "4.36.1"
7
+ app_file: app.py
8
+ pinned: false