How can i use my own yolov5 with my own backbone, Is it possible to do ?

kadirnar / Yolov7-SAHI

A lightweight vision library for performing large scale object detection & instance segmentation

MIT License

51 stars 14 forks source link

Open nakuda1 opened 1 year ago

nakuda1 commented 1 year ago

Hi sir, I do my own yolov5 with swintransformer backbone, if I want to use my own best.pt, what .py I need change. I am so confused, Thanks!

kadirnar commented 1 year ago

You need to write prediction codes. Just update the model.py file. You can open an issue to the original SAHI library.