Python load_and_split_all_point_clouds_under_folderの例

コード例 #1

ファイル: train_s_net.py プロジェクト: zhangguanghui1/learning_to_sample

project_dir = osp.dirname(osp.dirname(osp.abspath(__file__)))
top_in_dir = osp.join(project_dir, 'data',
                      'shape_net_core_uniform_samples_2048'
                      )  # Top-dir of where point-clouds are stored.
top_out_dir = osp.join(project_dir,
                       'results')  # Use to save Neural-Net check-points etc.

if flags.object_class == 'multi':
    class_name = ['chair', 'table', 'car', 'airplane']
else:
    class_name = [str(flags.object_class)]

# Load Point-Clouds
syn_id = snc_category_to_synth_id()[class_name[0]]
class_dir = osp.join(top_in_dir, syn_id)
pc_data_train, pc_data_val, _ = load_and_split_all_point_clouds_under_folder(
    class_dir, n_threads=8, file_ending='.ply', verbose=True)

for i in range(1, len(class_name)):
    syn_id = snc_category_to_synth_id()[class_name[i]]
    class_dir = osp.join(top_in_dir, syn_id)
    pc_data_train_curr, pc_data_val_curr, _ = load_and_split_all_point_clouds_under_folder(
        class_dir, n_threads=8, file_ending='.ply', verbose=True)
    pc_data_train.merge(pc_data_train_curr)
    pc_data_val.merge(pc_data_val_curr)

if flags.object_class == 'multi':
    pc_data_train.shuffle_data(seed=55)
    pc_data_val.shuffle_data(seed=55)

ae_dir = osp.join(top_out_dir, flags.ae_folder)

コード例 #2

ファイル: evaluate_s_net.py プロジェクト: zhangguanghui1/learning_to_sample

# Define basic parameters
project_dir = osp.dirname(osp.dirname(osp.abspath(__file__)))
top_in_dir = osp.join(project_dir, 'data',
                      'shape_net_core_uniform_samples_2048'
                      )  # Top-dir of where point-clouds are stored.
top_out_dir = project_dir  # Use to save Neural-Net check-points etc.

if flags.object_class == 'multi':
    class_name = ['chair', 'table', 'car', 'airplane']
else:
    class_name = [str(flags.object_class)]

# Load Point-Clouds
syn_id = snc_category_to_synth_id()[class_name[0]]
class_dir = osp.join(top_in_dir, syn_id)
_, _, pc_data_test = load_and_split_all_point_clouds_under_folder(
    class_dir, n_threads=8, file_ending='.ply', verbose=True)

for i in range(1, len(class_name)):
    syn_id = snc_category_to_synth_id()[class_name[i]]
    class_dir = osp.join(top_in_dir, syn_id)
    _, _, pc_data_test_curr = load_and_split_all_point_clouds_under_folder(
        class_dir, n_threads=8, file_ending='.ply', verbose=True)
    pc_data_test.merge(pc_data_test_curr)

# Load configuration
train_dir = osp.join(top_out_dir, flags.train_folder)
restore_epoch = 500
conf = Conf.load(osp.join(train_dir, 'configuration'))

conf.encoder_args['return_layer_before_symmetry'] = True