Open aiotko opened 2 years ago
If you can't convert to ONNX, try torch2trt
Thank you for the advice, I've tried but there are no expected properties in the model
import torch
from torch2trt import TRTModule
model_trt = TRTModule()
model_trt.load_state_dict(torch.load('rcnn_emd_refine.pth'))
def _load_from_state_dict(
...
engine_bytes = state_dict[prefix + "engine"]
...
self.input_names = state_dict[prefix + "input_names"]
self.output_names = state_dict[prefix + "output_names"]
Hi,
Sorry for one more silly question, I only started my way in machine learning and trying to integrate the CrowdDet model to be used with FastMOT. https://github.com/Purkialo/CrowdDet
I tried to convert it to onnx format but couldn't do it. Could you please give me a hint how to do and which parameters should be used for it?
Thank you very much in advance!