init

Author: Gaoyiminggithub

exp/inference/inference.py

@@ -6,7 +6,7 @@
 import os
 import sys
 from collections import OrderedDict
-sys.path.append('../../')
+sys.path.append('./')
 # PyTorch includes
 import torch
 from torch.autograd import Variable

inference.sh

@@ -0,0 +1 @@
+python exp/inference/inference.py --loadmodel ./data/pretrained_model/inference.pth --img_path ./img/messi.jpg --output_path ./img/ --output_name /output_file_name

networks/__init__.py

@@ -0,0 +1,3 @@
+from .deeplab_xception import *
+from .deeplab_xception_transfer import *
+from .deeplab_xception_universal import *