- 到github下载yolov8源码,得到ultralytics-main.zip,并解压。GitHub - ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite
- 数据集制作,将数据集文件夹放在与ultralytics-main同级目录下,(train文件夹和val文件夹下相同)如下图所示。
最后,在命令行cd到ultralytics-main文件夹下,运行命令
yolo classify train data=/opt/pycharm2021/workspace/BrightStripeDirectionModel/datasets/direction model=./yolov8s-cls.pt epochs=300 imgsz=640 amp=False