New Features

Classify

Train/Val/Predict with YOLOv5 image classifier:

$ yolov5 classify train --img 640 --data mnist2560 --weights yolov5s-cls.pt --epochs 1

$ yolov5 classify predict --img 640 --weights yolov5s-cls.pt --source images/

from yolov5.classify import train, val, predict

train.run(imgsz=640, data='mnist2560.yaml')
val.run(imgsz=640, weights='yolov5s-cls.pt')
predict.run(imgsz=640)

import yolov5

model = yolov5.load('yolov5s-cls.pt')

Segment

Train/Val/Predict with YOLOv5 instance segmentation model:

$ yolov5 segment train --img 640 --weights yolov5s-seg.pt --epochs 1

$ yolov5 segment predict --img 640 --weights yolov5s-seg.pt --source images/

from yolov5.segment import train, val, predict

train.run(imgsz=640, data='coco128.yaml')
val.run(imgsz=640, weights='yolov5s-seg.pt')
predict.run(imgsz=640)

import yolov5

model = yolov5.load('yolov5s-seg.pt')

What's Changed

• Fix pypi package version not updating in readme by @Isydmr in https://github.com/fcakyon/yolov5-pip/pull/141
• fix typo notebook utils by @fcakyon in https://github.com/fcakyon/yolov5-pip/pull/144
• update export arg usage by @fcakyon in https://github.com/fcakyon/yolov5-pip/pull/145
• update to ultralytics 16.09.22 by @fcakyon in https://github.com/fcakyon/yolov5-pip/pull/146
• add classify and segment usage into readme by @fcakyon in https://github.com/fcakyon/yolov5-pip/pull/147

New Contributors

• @Isydmr made their first contribution in https://github.com/fcakyon/yolov5-pip/pull/141

Full Changelog: https://github.com/fcakyon/yolov5-pip/compare/6.1.9...6.2.0