search

pytorch / glow

Compiler for Neural Network hardware accelerators
Apache License 2.0
3.23k stars 691 forks source link

[ModelZoo] support emotions_ferplus model #2504

Closed nickgg closed 5 years ago

nickgg commented 5 years ago

Add support for the FER+ Emotion Recognition model in glow (https://github.com/onnx/models/tree/master/emotion_ferplus) and verify via a runnable example we can include in the ModelZoo regression test.

The image format for this model is 64x64 0to1.

This issue blocks #2500.

arunm-git commented 5 years ago

@nickgg : is the image format really 0to1? Looking at https://github.com/onnx/models/tree/master/emotion_ferplus#preprocessing, it seems like 0to255, right? At least, when I run that preprocessing on a bunch of images, I get values that appear to lie within 0 to 255.

nickgg commented 5 years ago

Looks like it's not, sorry!