npoint=para_config_gan['point_cloud_shape'][0],
shuffle=True,
split='all',
preprocess=False)
else:
CLEAN_TRAIN_DATASET = shapenet_pc_dataset.ShapeNetPartPointsDataset(
para_config_gan['point_cloud_dir'],
batch_size=para_config_gan['batch_size'],
npoint=para_config_gan['point_cloud_shape'][0],
shuffle=True,
split='all',
preprocess=False)
NOISY_TRAIN_DATASET = shapenet_pc_dataset.ShapeNet_3DEPN_PointsDataset(
para_config_gan['3D-EPN_train_point_cloud_dir'],
batch_size=para_config_gan['batch_size'],
npoint=para_config_gan['point_cloud_shape'][0],
shuffle=True,
split='train',
preprocess=False)
NOISY_TEST_DATASET = shapenet_pc_dataset.ShapeNet_3DEPN_PointsDataset(
para_config_gan['3D-EPN_train_point_cloud_dir'],
batch_size=para_config_gan['batch_size'],
npoint=para_config_gan['point_cloud_shape'][0],
shuffle=False,
split='val',
preprocess=False)
#################### dirs, code backup and etc for this run ##########################
#LOG_DIR = os.path.join('run_3D-EPN', 'run_%s'%(cat_name), 'pcl2pcl', 'log_' + para_config_gan['exp_name'] + '_' + para_config_gan['loss'] + '_' + datetime.now().strftime('%Y-%m-%d-%H-%M-%S'))
LOG_DIR = os.path.join(
'run_3D-EPN', 'run_%s' % (cat_name), 'pcl2pcl',
model_idx)
elif pcl2pcl_mode == 'GT':
model_idx = 1000
para_config_gan[
'pcl2pcl_gan_ckpt'] = '/workspace/pcl2pcl-gan/pc2pc/run/run_3D-EPN_pcl2pcl_GT/run_dresser/log_dresser_pcl2pcl_GT_3D-EPN_emd_sharedAE_2019-08-06-18-42-35/ckpts/model_%s.ckpt' % (
model_idx)
if not os.path.exists(os.path.dirname(para_config_gan['pcl2pcl_gan_ckpt'])):
print('pcl2pcl_gan_ckpt not exist! %s' %
(os.path.dirname(para_config_gan['pcl2pcl_gan_ckpt'])))
exit()
NOISY_TEST_DATASET = shapenet_pc_dataset.ShapeNet_3DEPN_PointsDataset(
para_config_gan['3D-EPN_test_point_cloud_dir'],
batch_size=para_config_gan['batch_size'],
npoint=para_config_gan['point_cloud_shape'][0],
shuffle=False,
split=split_name,
preprocess=False)
SCAN_PC_DIR = config.ShapeNet_v1_point_cloud_dir
def prepare4test():
#################### dirs, code backup and etc for this run ##########################
model_name = para_config_gan['pcl2pcl_gan_ckpt'].split('/')[-1].split(
'.')[0]
#para_config_gan['LOG_DIR'] = os.path.join('results', 'test_3D-EPN_pcl2pcl_'+pcl2pcl_mode, 'test_%s'%(cat_name), split_name + '_' + para_config_gan['exp_name'] + '_' + model_name + '_' + datetime.now().strftime('%Y-%m-%d-%H-%M-%S'))
para_config_gan['LOG_DIR'] = os.path.join(
'results', 'test_3D-EPN_pcl2pcl_' + pcl2pcl_mode,
split_name + '_' + para_config_gan['exp_name'] + '_' + model_name +