Using yolov8 after training with Google Colab
Because of lack of data, there are only 7 classes: pikachu, charmander, bulbasaur, squirtle, eevee, jigglypuff and other.
pip install ultralytics
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.
Edit name.yaml.
Upload images and labels.
!yolo train model=yolov8n.pt data=/content/name.yaml epochs=50 imgsz=640
last.pt and best.pt in result folder.
Run predict.py to see result. This is my predict to test.mp4:
