this project doesn't support darknet yolov4 weights

NVIDIA-AI-IOT / yolo_deepstream

yolo model qat and deploy with deepstream&tensorrt

Apache License 2.0

547 stars 138 forks source link

this project doesn't support darknet yolov4 weights #5

Open futureflsl opened 3 years ago

futureflsl commented 3 years ago

pytorch-yolov4 postprocess is in model,but darknet yolov4 weights postprocess is not in model,inference and postprocess are splited,so what can we do?

mchi-zg commented 3 years ago

this yolov4 includes the postprocess. deepstream-yolov4 sample put the NMS is DS lib TRT-yolov4 sample includes NMS in the model