-
RetinaFace
-
CenterNet
-
DBNet
-
LFFD
-
YoloV3
-
SoloV2
-
SparseR-CNN
-
Yolact
-
knowledge distill
-
better image augmentation
1、onnx to tensorrt error:
Loading ONNX file from path epoch_89.onnx...
Beginning ONNX file parsing
Parser ONNX model failed.
In node 0 (importModel): INVALID_GRAPH: Assertion failed: tensors.count(input_name)
need change export script:
with torch.no_grad():
torch.onnx.export(
model,
inputs,
save_onnx_name,
export_params=True,
verbose=False,
training=False,
do_constant_folding=True,
keep_initializers_as_inputs=True, # ----> this line is very important
)