From dd03641429227f24ecd5adda34485f0d6d6e9c59 Mon Sep 17 00:00:00 2001 From: shiyu22 Date: Mon, 13 Dec 2021 19:30:58 +0800 Subject: [PATCH] Update README Signed-off-by: shiyu22 --- README.md | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 456c6b4..d19c589 100644 --- a/README.md +++ b/README.md @@ -10,7 +10,9 @@ This Operator generates feature vectors from the pytorch pretrained **Resnet50** ## Interface -`__init__(self, model_name: str, framework: str = 'pytorch')` +```python +__init__(self, model_name: str, framework: str = 'pytorch') +``` **Args:** @@ -21,7 +23,9 @@ This Operator generates feature vectors from the pytorch pretrained **Resnet50** - the framework of the model - supported types: `str`, default is 'pytorch' -`__call__(self, img_tensor: torch.Tensor)` +```python +__call__(self, img_tensor: torch.Tensor) +``` **Args:**