YOLOv3-tiny Pruning Recipe

[anujdutt9]

Hi, is there any recipe provided for pruning the YOLOv3-tiny model ? I could see the recipe for YOLOv3-SPP model but couldn't fine one for the tiny model.
Can you please share a recipe for that or provide a guide for making one for the tiny model?

Thanks.

[jeanniefinks]

Hi @anujdutt9
We are in the process of writing up a recipe approach (and testing!) per your post. We'll be in touch; shooting for this week.
Thank you! Best, Jeannie / Neural Magic

[markurtz]

Hi @anujdutt9, we've recently added an API to enable a simpler flow for creating recipes in our 0.9 SparseML release. This should address your needs and give you a great starting point for a recipe. To run it first export the YOLOv3-tiny model to ONNX and then use the following SparseML code to create the recipe:

from sparseml.sparsification import create_pruning_recipe

ONNX_FILE_PATH = "/PATH/TO/MODEL.onnx"
create_pruning_recipe(ONNX_FILE_PATH)