ViT
Collection
18 items • Updated
How to use mobilint/ViT_Base_Patch16_384 with Mobilint:
# pip install mblt-model-zoo
from mblt_model_zoo.vision import MBLT_Engine
model = MBLT_Engine(
model_cls="ViT_Base_Patch16_384",
model_type="DEFAULT",
model_path="",
core_mode="global8",
)
try:
image = model.preprocess("path/to/image.jpg")
output = model(image)
result = model.postprocess(output)
finally:
model.dispose()