Source of the models

[barak-hurwitz]

Hi,

Where can i find the model source (in original framework) used to create the ONNX model?
For Resnet50 ONNX model:
what PreProcessing need to be done on the images before i can feed them as input to the network as NPZ files. (PreProceesing on Image that we get from LMDB or BMP with values 0-255)

Thx,
Barak

[houseroad]

Right now, all the models are converted from Caffe2 model zoo.
https://github.com/caffe2/models

We are expending the model zoo, e.g., some models from PyTorch will be added.

[mx-iao]

Hi @barak-hurwitz , we will be adding some additional information to the READMEs of each model, including source code and information on the input data expected, in the upcoming weeks.

[barak-hurwitz]

Thanks,
Currently i assumed for Resnet50 that the range is -5=>5 (Based on the NPZ input min/max), so i did the scale and shift per channel and it worked when i converted BMP to the NPZ file.

[ankkhedia]

Hi @barak-hurwitz , please take a look into https://github.com/onnx/model-zoo. It contains some SOTA models with preprocessing codes. Please feel free to checkout the repo and raise an issue there in case you find something missing.

Thanks!

[barak-hurwitz]

Thanks :) , I'll check it out