search

yasenh / libtorch-yolov5

A LibTorch inference implementation of the yolov5
MIT License
372 stars 114 forks source link

could you pass a yolo5 weight file? i could download every one of them. #1

Closed xinsuinizhuan closed 3 years ago

yasenh commented 3 years ago

@xinsuinizhuan I use "YOLOv5s" model from https://github.com/ultralytics/yolov5, and you need to convert it to TorchScript file by https://github.com/ultralytics/yolov5/blob/master/models/export.py