CoAtNet
Collection
3 items • Updated
How to use mobilint/CoAtNet_2_RW_224 with Mobilint:
# pip install mblt-model-zoo
from mblt_model_zoo.vision import MBLT_Engine
model = MBLT_Engine(
model_cls="CoAtNet_2_RW_224",
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()
Base model
timm/coatnet_2_rw_224.sw_in12k_ft_in1k