d-li14/mobilenetv2.pytorch

图像预处理

phoenares opened this issue 5 years ago · 2 comments

phoenares commented 5 years ago

请问下使用模型的话，图像该怎么预处理啊，均值，方差，还有RGB通道，谢谢

d-li14 commented 5 years ago

refer to https://github.com/d-li14/mobilenetv2.pytorch/blob/master/imagenet.py#L197-L202, https://github.com/d-li14/mobilenetv2.pytorch/blob/master/imagenet.py#L214-L218

phoenares commented 5 years ago

refer to https://github.com/d-li14/mobilenetv2.pytorch/blob/master/imagenet.py#L197-L202, https://github.com/d-li14/mobilenetv2.pytorch/blob/master/imagenet.py#L214-L218

好的，谢谢