How to convert my yolov5s.pt to yolov5s.ptl

facebookresearch / playtorch

PlayTorch is a framework for rapidly creating mobile AI experiences.

https://playtorch.dev/

MIT License

830 stars 101 forks source link

How to convert my yolov5s.pt to yolov5s.ptl #206

Open Razamalik4497 opened 1 year ago

Razamalik4497 commented 1 year ago

Area Select

react-native-pytorch-core (core package)

Description

I have seen there is an export script https://playtorch.dev/docs/tutorials/prepare-custom-model/ for mobilenet, but I have trained yolov5s.pt how to export it for playtorch in yolov5s.ptl format.

smreo3839 commented 1 year ago

I have this problem, too https://github.com/ultralytics/yolov5/blob/master/export.py Export to yolov5.ptl option not found in yolov5 Therefore, it seems that an error occurs when loading the model with torch.jit._loadForMobile(filePath) Do you have a solution?