search

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.26k stars 4.18k forks source link

Add reparameterization for yolov7-tiny #2016

Open lweicker opened 5 months ago

dsbyprateekg commented 5 months ago

I tried your code on the custom yolov7-tiny model. And its size is reduced to 11.8 MB only from 11.9 MB.

lweicker commented 5 months ago

@dsbyprateekg Once reparameterization is done, you can convert directly the model to engine file, for example in int8, and the output file will be about 8.3MB.

dsbyprateekg commented 5 months ago

@dsbyprateekg Once reparameterization is done, you can convert directly the model to engine file, for example in int8, and the output file will be about 8.3MB.

any link or tutorial for doing this conversion?

lweicker commented 5 months ago

@dsbyprateekg sure, you can check: https://github.com/wang-xinyu/tensorrtx/tree/master/yolov7