ceccocats / tkDNN

Deep neural network library and toolkit to do high performace inference on NVIDIA jetson platforms
GNU General Public License v2.0
718 stars 208 forks source link

Problems with custom dataset and tensorrt8 #301

Open Toufengmehmet09 opened 1 year ago

Toufengmehmet09 commented 1 year ago

Hello i have an issue using a custom trained dataset with one class on yolov4. Everything works fine on tensorrt7 but when using tensorrt8 i get 3-4 detections around the object as it gets bigger. I tested the provided weights, both with tensorrt7 and tensorrt8 there is no problem. What could be the issue? I attached the yolo-config file. yolov4-custom.txt