models/__init__.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Sun Mar 20 14:23:55 2022

@author: jma
"""

from .unetr2d import UNETR2D
from .swin_unetr import SwinUNETR