megvii-model/ShuffleNet-Series

Will you release Pretrained Model based on RGB and ToTensor() Inputs ?

Alexandmrwh opened this issue · 2 comments

Great works! The pretrained model you release is based on bgr input (0-255), will you release the rgb input (transformed by pytorch) ?

Why don't use the ToTensor() in PyTorch?

use ToBGRTensor() and OpenCVReize() for much faster ? or for better accuracy ?

nmaac commented

Hi @Alexandmrwh , the models we trained are based on BGR input, you may need to adjust your input format if you use the pretrained models.

Hi @zhjpqq , it doesn't affect speed and accuracy, we use it just to keep the data format consistent with our inner framework's format.