use_gt_info = True ## others num_thread = 20 gpu_ids = '0' num_gpus = 1 continue_train = False def set_args(self, gpu_ids, continue_train=False): self.gpu_ids = gpu_ids self.num_gpus = len(self.gpu_ids.split(',')) self.continue_train = continue_train os.environ["CUDA_VISIBLE_DEVICES"] = self.gpu_ids print('>>> Using GPU: {}'.format(self.gpu_ids)) cfg = Config() sys.path.insert(0, osp.join(cfg.root_dir, 'common')) from utils.dir_utils import add_pypath, make_folder add_pypath(osp.join(cfg.data_dir)) for i in range(len(cfg.trainset_3d)): add_pypath(osp.join(cfg.data_dir, cfg.trainset_3d[i])) for i in range(len(cfg.trainset_2d)): add_pypath(osp.join(cfg.data_dir, cfg.trainset_2d[i])) add_pypath(osp.join(cfg.data_dir, cfg.testset)) make_folder(cfg.model_dir) make_folder(cfg.vis_dir) make_folder(cfg.log_dir) make_folder(cfg.result_dir)
load_path=None, test=None, params=None): self.dataset = dataset self.model = model self.feats = feats self.save_path = save_path self.continue_train = continue_train self.load_path = load_path self.test = test self.params = params #Do not comment "cfg" as it is a global configuration object that will be called later. cfg = Config() # sys.path.insert(0, cfg.root_dir) # sys.path.insert(0, os.path.join(cfg.root_dir, 'data')) from utils.dir_utils import add_pypath, make_folder # add_pypath(os.path.join(cfg.data_dir)) # for i in range(len(cfg.datasets)): # add_pypath(os.path.join(cfg.data_dir, cfg.datasets[i])) # Create the folder to store the best trained model make_folder(cfg.model_dir) # make_folder(cfg.vis_dir) # Create the folder to store the log file make_folder(cfg.log_dir) # make_folder(cfg.result_dir)