--- a
+++ b/main.py
@@ -0,0 +1,11 @@
+from utils import config
+GPU_LIST = config.INFERENCE_GPUS
+import os
+os.environ["CUDA_VISIBLE_DEVICES"] = ','.join('{0}'.format(n) for n in GPU_LIST)
+from inference import Inference
+
+
+if __name__ == '__main__':
+    pg = Inference(data_dir='/path/to/data/', data_list='/path/to/list',
+                    class_num=2, result_dir='./result', use_level=1)
+    pg.run()