English 中文(简体)
Convert pytorch trained model .pt file into a .wts plaintext file
原标题:

I have a custom pretrained pytorch model and I need to convert it into a .wts file to run a library on it but I am lost as to how to accomplish this.

Details: I used YOLOv5 to train a model and save it as a .pt file. I need to run this model on Jetson Nano and the library I am using takes in a .wts file extension. I found only one website referring to this conversion here: https://github.com/wang-xinyu/tensorrtx/blob/master/tutorials/getting_started.md however, the README does not explain how to actually accomplish this.

Any help will be appreciated!

Things I have tried:

  • I used YOLOv5 to convert the .pt model into a ONNX model.I am not sure how to convert the ONNX model into a .wts file.
  • I tried using trtexec to convert the file into something TensorRT can use but that converts it into a .trt file.
  • I tried using pytorch2trt but again that is accomplishing something else
问题回答

暂无回答




相关问题
Resample Filter of WEKA - How to interpret the result

I am currently strugeling with a machine learning problem whereas I have to deal with great unbalanced data sets. That is, there are six classes ( 1 , 2 ... 6 ). Unfortunately there are e.g. for class ...

How to recognize rectangles in this image?

I have a image with horizontal and vertical lines. In fact, this image is the BBC website converted to horizontal and vertical lines. My problem is that I want to be able to find all the rectangles in ...

Question About Using Weka, the machine learning tool

I m using the explorer feature of Weka for classification. So I have my .arff file, with 2 features of NUMERIC value, and my class is a binary 0 or 1 (eg {0,1}). Sample: @RELATION summary @...

Implementing a linear, binary SVM (support vector machine)

I want to implement a simple SVM classifier, in the case of high-dimensional binary data (text), for which I think a simple linear SVM is best. The reason for implementing it myself is basically that ...

libsvm model file format

According to this FAQ the model format in libsvm should be straightforward. And in fact it is, when I call just svm-train. As an example, the first SV for the a1a dataset is 1 3:1 11:1 14:1 19:1 39:...

Competitive Learning in Neural Networks

I am playing with some neural network simulations. I d like to get two neural networks sharing the input and output nodes (with other nodes being distinct and part of two different routes) to compete. ...

热门标签