I have a .pt segmentation model(yolo v5) and I want to create .wts and .cfg file from that would you share any resources for it?
I have .pt segmentaion model(yolo v5) and I want to create .wts and .cfg file along with .so file. Could anyone share the resource for that?
There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks
Seems to be a PyTorch question, just made a quick search and found something something below, not sure if this is what you are looking for:
Script: tensorrtx/gen_wts.py at master · wang-xinyu/tensorrtx · GitHub
Usage: python gen_wts.py -w YOUR-PT-FILE.pt -o CONVERTED-PTS-FILE.wts
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.