![]() YOLOv5 segmentation training supports auto-download COCO128-seg segmentation dataset with -data coco128-seg.yaml argument and manual download of COCO-segments dataset with bash data/scripts/get_coco.sh -train -val -segments and then python train.py -data coco.yaml. Reproduce by python export.py -weights yolov5s-seg.pt -include engine -device 0 -half Export to ONNX at FP32 and TensorRT at FP16 done with export.py.Reproduce by python segment/val.py -data coco.yaml -weights yolov5s-seg.pt -batch 1 Values indicate inference speed only (NMS adds about 1ms per image). ![]()
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |