search

nvdla / hw

RTL, Cmodel, and testbench for NVDLA
Other
1.72k stars 567 forks source link

CNN models for nv_small #360

Open mtsanic opened 2 years ago

mtsanic commented 2 years ago

Hi, I want to test nv_small on different CNN models. So far, I was able to find models for LeNet and ResNet-18. I found a ResNet-50 before, but it was missing calibration table. Does anybody have caffemodel, prototxt and calibration table for any other models for nv_small?

Any help would be appreciated!

tirumalnaidu commented 2 years ago

You can find the calibration table for resnet50 here: https://github.com/nvdla/sw/tree/master/umd/utils/calibdata Caffe model files and loadables for NVDLA can be found here: Link Even Yolox-nano object detection has been ported to NVDLA using tengine: Link