detectron2 algorithm extension

RetinaFace

CenterNet

TODO

• RetinaFace

• CenterNet

• DBNet

• LFFD

• YoloV3

• SoloV2

• SparseR-CNN

• Yolact

• knowledge distill

• better image augmentation

Issus

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
    )

Acknowledgement

Detectron2

CenterX

RetinaFace

PaddleOCR