(
config_dict: T.Dict[str, T.Any],
dataset_name: str,
total: int = 10000,
max_angle: float = 180,
# max_translate_ratio: float = 6.0,
# ray_perturbation_angle: float = 0.75,
width_px: int = 300,
height_px: int = 300,
target_width_px: int = 80,
target_height_px: int = 80,
k: int = 40,
local_max_angle: float = 3.,
n_imgs: int = 30,
min_subsample: int = 1,
max_subsample: int = 30,
min_k_ratio: float = 1.,
max_k_ratio: float = 1.,
mesh_scale: float = 1.,
min_r: float = 0.5, # min_r used for output camera
max_r: float = 3., # max_r used for output camera
rand_r: float = 0.,
texture_mode: str = 'ori', # 'files', 'imagenet'
texture_crop_method: T.Union[int, str] = 'ori', # or an int p indiciating the min p * p crop
texture_filenames: T.List[str] = None,
render_method: str = 'ray_cast', # 'rasterization'
num_threads: int = 10,
use_bucket_sampler: bool = True,
mix_meshes: bool = False,
min_num_mesh: int = 1,
max_num_mesh: int = 2,
radius_scale: float = 2.,
total_combined: int = None,
)
| 74 | |
| 75 | |
| 76 | def set_dataset( |
| 77 | config_dict: T.Dict[str, T.Any], |
| 78 | dataset_name: str, |
| 79 | total: int = 10000, |
| 80 | max_angle: float = 180, |
| 81 | # max_translate_ratio: float = 6.0, |
| 82 | # ray_perturbation_angle: float = 0.75, |
| 83 | width_px: int = 300, |
| 84 | height_px: int = 300, |
| 85 | target_width_px: int = 80, |
| 86 | target_height_px: int = 80, |
| 87 | k: int = 40, |
| 88 | local_max_angle: float = 3., |
| 89 | n_imgs: int = 30, |
| 90 | min_subsample: int = 1, |
| 91 | max_subsample: int = 30, |
| 92 | min_k_ratio: float = 1., |
| 93 | max_k_ratio: float = 1., |
| 94 | mesh_scale: float = 1., |
| 95 | min_r: float = 0.5, # min_r used for output camera |
| 96 | max_r: float = 3., # max_r used for output camera |
| 97 | rand_r: float = 0., |
| 98 | texture_mode: str = 'ori', # 'files', 'imagenet' |
| 99 | texture_crop_method: T.Union[int, str] = 'ori', # or an int p indiciating the min p * p crop |
| 100 | texture_filenames: T.List[str] = None, |
| 101 | render_method: str = 'ray_cast', # 'rasterization' |
| 102 | num_threads: int = 10, |
| 103 | use_bucket_sampler: bool = True, |
| 104 | mix_meshes: bool = False, |
| 105 | min_num_mesh: int = 1, |
| 106 | max_num_mesh: int = 2, |
| 107 | radius_scale: float = 2., |
| 108 | total_combined: int = None, |
| 109 | ) -> T.Dict[str, T.Any]: |
| 110 | config_dict['dataset_info']['dataset_name'] = dataset_name |
| 111 | mesh_filename, test_mesh_filename, dataset_root_dir = get_mesh_filenames(dataset_name) |
| 112 | config_dict['dataset_info']['dataset_root_dir'] = dataset_root_dir |
| 113 | config_dict['dataset_info']['mesh_filename'] = mesh_filename |
| 114 | config_dict['dataset_info']['test_mesh_filename'] = test_mesh_filename |
| 115 | config_dict['dataset_info']['total'] = total |
| 116 | config_dict['dataset_info']['max_angle'] = max_angle |
| 117 | # config_dict['dataset_info']['max_translate_ratio'] = max_translate_ratio |
| 118 | # config_dict['dataset_info']['ray_perturbation_angle'] = ray_perturbation_angle |
| 119 | config_dict['dataset_info']['width_px'] = width_px |
| 120 | config_dict['dataset_info']['height_px'] = height_px |
| 121 | config_dict['dataset_info']['target_width_px'] = target_width_px |
| 122 | config_dict['dataset_info']['target_height_px'] = target_height_px |
| 123 | |
| 124 | config_dict['dataset_info']['k'] = k |
| 125 | config_dict['dataset_info']['local_max_angle'] = local_max_angle |
| 126 | config_dict['dataset_info']['n_imgs'] = n_imgs |
| 127 | |
| 128 | config_dict['dataset_info']['min_subsample'] = min_subsample |
| 129 | config_dict['dataset_info']['max_subsample'] = max_subsample |
| 130 | config_dict['dataset_info']['min_k_ratio'] = min_k_ratio |
| 131 | config_dict['dataset_info']['max_k_ratio'] = max_k_ratio |
| 132 | config_dict['dataset_info']['mesh_scale'] = mesh_scale |
| 133 | config_dict['dataset_info']['min_r'] = min_r |
no test coverage detected