search

Xilinx / finn-examples

Dataflow QNN inference accelerator examples on FPGAs
BSD 3-Clause "New" or "Revised" License
164 stars 56 forks source link

ResNet-50 Python Brevitas Model #44

Open oshears opened 1 year ago

oshears commented 1 year ago

Hi FINN Team,

I am currently working on a project that is interested in recreating the quantized .onnx file used for the ResNet-50 FINN example here:

https://github.com/Xilinx/finn-examples/tree/main/build/resnet50

Is there any way that the notebook/python code for creating the quantized, brevitas version of the ResNet-50 model can be shared?

I am interested in trying to adapt the network to fit on a ZCU102/ZCU104.

mmrahorovic commented 1 year ago

Hi @oshears, good question and great to hear about your project! The Brevitas code for ResNet-50 is indeed currently not released, but we are working on it and we will let you know in the upcoming months when it is ready!

MRFiruzabadi commented 9 months ago

Hi @mmrahorovic, Did you manage to release the Brevitas code for ResNet-50?

mmrahorovic commented 9 months ago

Hi @MRFiruzabadi The code isn't yet released. However, in the meantime, you can find the ResNet-18 model in Brevitas, which can serve as a starting point for extending this to a ResNet-50 model: https://github.com/Xilinx/brevitas/blob/dev/src/brevitas_examples/bnn_pynq/models/resnet.py.

Adib6637 commented 2 months ago

Hi, has the code for resnet 50 been released?