Yolov4 TRT gives error during evaluate mAP

[pangshengwei]

Hi, I tried training on the example mnist dataset provided but am getting this error when trying to evaluate the mAP. Seems like the code expects Yolo(batched_input) to return a dict but its returning a list of numpy arrays, Im wondering what could be the problem? I have attached the screenshot of the error and my config, any help would be appreciated thanks!

Update 1: This issue only occurs with TRT is used, it works OK with regular TF

Update 2: I tried setting pred_bbox = Yolo(batched_input) just like how its done for TF, and it seems to fix the issue, please advise on wether this is correct?

[pythonlessons]

Not sure why you get that error, for me everything was fine