зеркало из https://github.com/microsoft/hi-ml.git
22 строки
593 B
Python
22 строки
593 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_resnet 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
|