v6.2.0
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 #141
- fix typo notebook utils by @fcakyon in #144
- update export arg usage by @fcakyon in #145
- update to ultralytics 16.09.22 by @fcakyon in #146
- add classify and segment usage into readme by @fcakyon in #147
New Contributors
Full Changelog: 6.1.9...6.2.0