The yolo-csp custom training not detect anything when converted to trt.

talebolano / TensorRT-Scaled-YOLOv4

TensorRT for Scaled YOLOv4(yolov4-csp.cfg)

10 stars 4 forks source link

The yolo-csp custom training not detect anything when converted to trt. #11

Open long-senpai opened 3 years ago

long-senpai commented 3 years ago

HI, as the topic name, i have a custom trained yolov4-csp with 7 classes, when i converted the .weight -> .trr model. I got success but when test some images. The trt model cannot detect anything. I have modified the config.h by my classes=7 and the classes_name respective. What im i wrong here?

long-senpai commented 3 years ago

you can get the .cfg file here

long-senpai commented 3 years ago

I was tested with the pretrained yoloV4-CSP model-> Trt model and it's work normally. So confuse!!