1 changed files with
3 additions and
3 deletions
-
README.md
|
|
@ -10,17 +10,17 @@ This Operator generates feature vectors from the pytorch pretrained **Resnet50** |
|
|
|
|
|
|
|
|
## Interface |
|
|
## Interface |
|
|
|
|
|
|
|
|
`Class Resnet50ImageEmbedding(Operator)` [source](./resnet50_image_embedding.py) |
|
|
|
|
|
|
|
|
`Class Resnet50ImageEmbedding(Operator)` [\[source\]](./resnet50_image_embedding.py) |
|
|
|
|
|
|
|
|
`__init__(self, model_name: str)` |
|
|
`__init__(self, model_name: str)` |
|
|
|
|
|
|
|
|
**param:** |
|
|
|
|
|
|
|
|
**params:** |
|
|
|
|
|
|
|
|
- model_name(str): the model name for embedding, like 'resnet50'. |
|
|
- model_name(str): the model name for embedding, like 'resnet50'. |
|
|
|
|
|
|
|
|
`__call__(self, img_tensor: torch.Tensor)` |
|
|
`__call__(self, img_tensor: torch.Tensor)` |
|
|
|
|
|
|
|
|
**param:** |
|
|
|
|
|
|
|
|
**params:** |
|
|
|
|
|
|
|
|
- img_tensor(torch.Tensor): the normalized image tensor. |
|
|
- img_tensor(torch.Tensor): the normalized image tensor. |
|
|
|
|
|
|
|
|
|
