tensorrtx-yolov7 icon indicating copy to clipboard operation
tensorrtx-yolov7 copied to clipboard

Incorrect result of transformation inference when I train a custom dataset model

Open JunGenius opened this issue 2 years ago • 2 comments

hello, author When I retrain, changing the number of classes, the result of inference is incorrect. But using official model inference, the result is correct. Can you help me, how to solve this problem? Thanks.

JunGenius avatar Sep 03 '22 03:09 JunGenius

hello, author When I retrain, changing the number of classes, the result of inference is incorrect. But using official model inference, the result is correct. Can you help me, how to solve this problem? Thanks.

maybe you should reparameterization for exmple if you train yolov7.pt with your datasets , you need reparameterization yolov7.pt , you can see https://github.com/WongKinYiu/yolov7

QIANXUNZDL123 avatar Sep 06 '22 05:09 QIANXUNZDL123

Yeah, after I reparameterization, the results are normal. thanks.

JunGenius avatar Sep 11 '22 02:09 JunGenius