deeprobust/image/netmodels/__init__.py

#__init__.py
from deeprobust.image.netmodels import CNN
from deeprobust.image.netmodels import resnet
from deeprobust.image.netmodels import YOPOCNN
from deeprobust.image.netmodels import train_model

__all__ = ['CNNmodel','resnet','YOPOCNN','train_model']