|
def | __init__ (self, gpu_id=None) |
|
def | __init__ (self, task, model_scale) |
|
def | construct_sample (self, image, text) |
|
def | encode_text (self, text, length=None, append_bos=False, append_eos=False) |
|
def | infer (self, img, text) |
|
def | infer (self, img, texts) |
|
def | visual_grounding (self, Image, Text) |
|
Definition at line 19 of file clip/server.py.
◆ __init__() [1/2]
def server.Inference.__init__ |
( |
|
self, |
|
|
|
gpu_id = None |
|
) |
| |
◆ __init__() [2/2]
def server.Inference.__init__ |
( |
|
self, |
|
|
|
task, |
|
|
|
model_scale |
|
) |
| |
◆ construct_sample()
def server.Inference.construct_sample |
( |
|
self, |
|
|
|
image, |
|
|
|
text |
|
) |
| |
◆ encode_text()
def server.Inference.encode_text |
( |
|
self, |
|
|
|
text, |
|
|
|
length = None , |
|
|
|
append_bos = False , |
|
|
|
append_eos = False |
|
) |
| |
◆ infer() [1/2]
def server.Inference.infer |
( |
|
self, |
|
|
|
img, |
|
|
|
text |
|
) |
| |
◆ infer() [2/2]
def server.Inference.infer |
( |
|
self, |
|
|
|
img, |
|
|
|
texts |
|
) |
| |
◆ visual_grounding()
def server.Inference.visual_grounding |
( |
|
self, |
|
|
|
Image, |
|
|
|
Text |
|
) |
| |
◆ cfg
◆ device
◆ generator
server.Inference.generator |
◆ gpu_id
◆ pad_idx
◆ patch_resize_transform
server.Inference.patch_resize_transform |
◆ preprocess
server.Inference.preprocess |
◆ task
◆ task_name
server.Inference.task_name |
◆ use_cuda
server.Inference.use_cuda |
◆ use_fp16
server.Inference.use_fp16 |
The documentation for this class was generated from the following file: