Skip to content

Conversation

kennybradley
Copy link
Contributor

I saw that yolov7-tiny within https://github.com/nihui/ncnn-assets/tree/master/models so it could be available to the python interface if there was a wrapper that executed the network call and post processing.

I added the yolov7.py file and updated the model_zoo and model_store so that it can transparently run within the python interface with model_zoo.

I followed the format of the other model_zoo entries as closely as I could.

I added an optional argument use_strides to the arguments that restricts which strides to extract from the network in case users know ahead of time the size of the objects they wish to detect. Since the anchor sizes are strongly connected to the size of the objects detected you can speed up yolov7 by targeting just the output layers that you are interested in.

Adding yolov7_Tiny to the model zoo list
Adding yolov7-tiny to the model_store since the param and bin files already exist in https://github.com/nihui/ncnn-assets/raw/master/models/
@tencent-adm
Copy link
Member

tencent-adm commented May 4, 2023

CLA assistant check
All committers have signed the CLA.

@nihui nihui merged commit f79daf4 into Tencent:master May 8, 2023
@nihui
Copy link
Member

nihui commented May 8, 2023

Thanks for your contribution !

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants