зеркало из https://github.com/microsoft/hi-ml.git
24 строки
602 B
Python
24 строки
602 B
Python
# autopep8: off
|
|
dependencies = ["torch", "torchvision"]
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
repo_dir = Path(__file__).parent
|
|
multimodal_src_dir = repo_dir / "hi-ml-multimodal" / "src" / "health_multimodal" / "image"
|
|
sys.path.append(str(multimodal_src_dir))
|
|
|
|
from model import ImageModel
|
|
from model import get_biovil_image_encoder as _biovil_resnet
|
|
|
|
# autopep8: on
|
|
|
|
|
|
def biovil_resnet(pretrained: bool = False) -> ImageModel:
|
|
"""Get BioViL image encoder.
|
|
|
|
:param pretrained: Load pretrained weights from a checkpoint.
|
|
"""
|
|
model = _biovil_resnet(pretrained=pretrained)
|
|
return model
|