提交 a71f68dd 编写于 作者: M MRXLT

fix imagenet demo

上级 fb0f6612
...@@ -30,7 +30,7 @@ else: ...@@ -30,7 +30,7 @@ else:
class ImageService(WebService): class ImageService(WebService):
def init_imagenet_setting(self): def init_imagenet_setting(self):
self.seq = Sequential([ self.seq = Sequential([
Base64ToImage(), Resize(256), CenterCrop(224), RGB2BGR(), Transpose( URL2Image(), Resize(256), CenterCrop(224), RGB2BGR(), Transpose(
(2, 0, 1)), Div(255), Normalize([0.485, 0.456, 0.406], (2, 0, 1)), Div(255), Normalize([0.485, 0.456, 0.406],
[0.229, 0.224, 0.225], True) [0.229, 0.224, 0.225], True)
]) ])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册