thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.13k stars 2.08k forks source link

Yolov2 Custom Weight/Ckpt/Meta to TensorRT #1125

Open tonginleong opened 4 years ago

tonginleong commented 4 years ago

Hello, Maybe a too-personal question but a desperate call here.

I started my project with yolov2 a while ago and had working/satisfying custom-trained ckpt/meta pairs to detect what I want. I wasn't able to save the training result as .weight files but the ckpt/meta worked just fine (I started the training with a modified .cfg file and a corresponding pretrained .weight file and used the "flow -train" command to point out the directories).

Then, v3 came out but I don't see there is an immediate need to move onto v3 since my v2 was working.

(Beside getting high-end/top-tier hardware), about a month ago I found out that Nvidia has TensorRT that could potentially speed up my program a lot, which is a very important task), and I have been doing my researches since then. But since v3 came out, I have troubles finding helpful posts for v2 out there (I also don't really find the v3 posts helpful even as a reference because the ones I found are not completed/too general in my view).

Much appreciated if someone has the same kind of experience and can provide me some guides/hints/steps on my way of "conversion".