search

lewes6369 / TensorRT-Yolov3

TensorRT for Yolov3
MIT License
487 stars 165 forks source link

Detected object in TensorRT VS Darknet model #44

Closed Rahn80643 closed 5 years ago

Rahn80643 commented 5 years ago

Hi, when I convert my YOLOv3 model to tensorRT, some objects that can be detected in YOLOv3 are not detected in TensorRT with FP32 mode.

Objects detected in Darknet YOLOv3 image

Objects detected in TensorRT YOLOv3 image

Should I modify the training data or modify some settings in TensorRT ?

lewes6369 commented 5 years ago

Is this your own model? please try to check the yolo config in "tensorRTWrapper/code/include/YoloConfigs.h" and the nms threshold.

Rahn80643 commented 5 years ago

I solved this by modifying the training data, and retrain the model. Thank you.