TPH-YOLOv5 Model Prediction and Evaluation

[bshakya77]

Hello @fcakyon,

I would like to know if this repository allows the prediction and evaluation using TPH-YOLOv5 model:

It is an improved version built on the top of YOLOv5 model. I am trying to use TPH-YOLOv5 model trained on VisDrone2019 dataset with SAHI, but when I use the checkpoint, it give me the following error:

File "/lib/python3.11/site-packages/sahi/models/yolov5.py", line 32, in load_model
    raise TypeError("model_path is not a valid yolov5 model path: ", e)
TypeError: ('model_path is not a valid yolov5 model path: ', AttributeError("Can't get attribute 'C3STR' on <module 'models.common'>

I used both YOLOv5 and ultralytics as model_type. But, it did not worked out. If you have any idea how to predict and evaluate using TPH-YOLOv5 model, it would be very helpful.

I appreciate your help.

Regards,
Bijay