def main(argv=None): if not gfile.Exists(COARSE_DIR): gfile.MakeDirs(COARSE_DIR) if not gfile.Exists(REFINE_DIR): gfile.MakeDirs(REFINE_DIR) if(TEST): test() elif(TRAIN): train()