最近沉迷炼丹, 效果图:
框架Ultralytics YOLOv8
来自GitHub的介绍:
Ultralytics YOLOv8 is a cutting-edge, state-of-the-art (SOTA) model that builds upon the success of previous YOLO versions and introduces new features and improvements to further boost performance and flexibility. YOLOv8 is designed to be fast, accurate, and easy to use, making it an excellent choice for a wide range of object detection and tracking, instance segmentation, image classification and pose estimation tasks.
具体的使用在官网有介绍,不再赘述。
笔者使用GPU NVIDIA GeForce RTX 3090, 24260MiB进行的训练和预测,尝试使用cpu训练过,将batch_size调整到1-2也可以训练,再大的话,就要看机器的性能了。
废话不多说,直接进行炼丹环节。
注意:使用v7的话,标注会出现中文乱码,所以就用v8