def _create_original_density_maps(self, force_create_den_maps):
if not osp.exists(self.ori_dir_den):
os.makedirs(self.ori_dir_den)
elif not force_create_den_maps:
return
create_density_map(self.ori_dir_img,
self.ori_dir_lab,
self.ori_dir_den,
mode=self.metadata['gt_mode'])
def _create_original_density_maps(self, force_create_den_maps):
all_density_dirs_exist = osp.exists(self.ori_dir_partA_train_den)
all_density_dirs_exist = all_density_dirs_exist and osp.exists(
self.ori_dir_partA_test_den)
all_density_dirs_exist = all_density_dirs_exist and osp.exists(
self.ori_dir_partB_train_den)
all_density_dirs_exist = all_density_dirs_exist and osp.exists(
self.ori_dir_partB_test_den)
if not all_density_dirs_exist:
mkdir_if_missing(self.ori_dir_partA_train_den)
mkdir_if_missing(self.ori_dir_partA_test_den)
mkdir_if_missing(self.ori_dir_partB_train_den)
mkdir_if_missing(self.ori_dir_partB_test_den)
elif not force_create_den_maps:
return
create_density_map(self.ori_dir_partA_train_img,
self.ori_dir_partA_train_lab,
self.ori_dir_partA_train_den,
self.ori_dir_partA_train_fac,
mode=self.metadata['gt_mode'])
create_density_map(self.ori_dir_partA_test_img,
self.ori_dir_partA_test_lab,
self.ori_dir_partA_test_den,
self.ori_dir_partA_test_fac,
mode=self.metadata['gt_mode'])
create_density_map(self.ori_dir_partB_train_img,
self.ori_dir_partB_train_lab,
self.ori_dir_partB_train_den,
self.ori_dir_partB_train_fac,
mode=self.metadata['gt_mode'])
create_density_map(self.ori_dir_partB_test_img,
self.ori_dir_partB_test_lab,
self.ori_dir_partB_test_den,
self.ori_dir_partB_test_fac,
mode=self.metadata['gt_mode'])
def _create_original_density_maps(self, force_create_den_maps):
mkdir_if_missing(self.ori_train_den)
mkdir_if_missing(self.ori_val_den)
mkdir_if_missing(self.ori_test_den)
if len(os.listdir(self.ori_train_den)) != len(
os.listdir(self.ori_train_gt)):
create_density_map(self.ori_train_img,
self.ori_train_lab,
self.ori_train_den,
self.ori_train_fac,
mode=self.metadata["gt_mode"])
if len(os.listdir(self.ori_val_den)) != len(os.listdir(
self.ori_val_gt)):
create_density_map(self.ori_val_img,
self.ori_val_lab,
self.ori_val_den,
self.ori_val_fac,
mode=self.metadata["gt_mode"])
if len(os.listdir(self.ori_test_den)) != len(
os.listdir(self.ori_test_gt)):
create_density_map(self.ori_test_img,
self.ori_test_lab,
self.ori_test_den,
self.ori_test_fac,
mode=self.metadata["gt_mode"])