--- a
+++ b/rvseg/models/__init__.py
@@ -0,0 +1,3 @@
+from .convunet import unet
+from .dilatedunet import dilated_unet
+from .dilateddensenet import dilated_densenet, dilated_densenet2, dilated_densenet3