上传者: 42134878
|
上传时间: 2021-10-26 18:55:31
|
文件大小: 188KB
|
文件类型: -
在TorchServe上运行的Yolov5(与GPU兼容)!
这是一个用于为Yolo v5对象检测模型运行TorchServe的dockerfile。 (TorchServe(PyTorch库)是一种灵活且易于使用的工具,用于服务从PyTorch导出的深度学习模型)。
您只需要在ressources文件夹中传递一个yolov5权重文件(.pt),它将部署一个http服务器,准备进行预测。
设置Docker映像
如果使用GPU,则在本地构建Torchserve映像(dockerhub一个错误): Build the image torchserve locally for GPU before running this (cf github torchserve) https://github.com/pytorch/serve/tree/master/docker
注意:仅对于CPU