search

Chris-hughes10 / Yolov7-training

A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the model on custom (non-COCO) tasks.
GNU General Public License v3.0
116 stars 35 forks source link

Add yolov7-tiny pretrained weights #14

Closed tkupek closed 1 year ago

tkupek commented 1 year ago

Weights file has to be added to releases manually. If necessary, bump release version in the urls.

The weights can be found here.

Chris-hughes10 commented 1 year ago

Hi @tkupek, thanks for the PR. I have updated the release to include the new state dict, please can you confirm that this works with the updated release and then we can merge

tkupek commented 1 year ago

I am not available for the next week, but will update once I'm back.

tkupek commented 1 year ago

@Chris-hughes10 the url worked for me. Ready to merge.

Chris-hughes10 commented 1 year ago

Hi @tkupek, following the discussion in the issue here it seems that a minor code change is required to use the tiny weights. Would you be able to update your PR to reflect this?

From what I can see, this could be done by setting a flag in the Detection head which controls whether or not to use the implicit modules. For all of the other models, this could default to True, but could be disabled for the tiny weights

tkupek commented 1 year ago

@Chris-hughes10 no problem at all, I have already adjusted my own version of the repo. Pushed the config change just now, please cheack.

Chris-hughes10 commented 1 year ago

Excellent, thanks for the speedy turnaround! The only minor point I see is that the state dict attached to the release is yolov7-tiny_state_dict.pt not yolov7-tiny_training_state_dict.pt. If you can update that, we are ready to merge I think

tkupek commented 1 year ago

Thanks for the hint. The string is updated.

Chris-hughes10 commented 1 year ago

Merged, thanks for your work on this :)