Just a side question:
Now that the pruning code runs without any errors.
How do I know that it actually works ? Do I need to implement some bounding box detection mechanism such as YOLO or Fast-RCNN ? Could you comment about this ?