本文整理汇总了Python中util.get_args方法的典型用法代码示例。如果您正苦于以下问题：Python util.get_args方法的具体用法？Python util.get_args怎么用？Python util.get_args使用的例子？那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类util的用法示例。
在下文中一共展示了util.get_args方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: main
# 需要导入模块: import util [as 别名]
# 或者: from util import get_args [as 别名]
def main():
config = util.get_args()
config.cuda = not config.no_cuda
torch.cuda.set_device(config.gpu)
if config.cuda and torch.cuda.is_available():
print("Using CUDA on GPU ", config.gpu)
else:
print("Not using CUDA.")
net = partnet_model.PARTNET(config)
net.load_state_dict(torch.load(config.save_path + '/partnet_final.pkl', map_location=lambda storage, loc: storage.cuda(config.gpu)))
if config.cuda:
net.cuda()
net.eval()
if not os.path.exists(config.output_path + 'segmented'):
os.makedirs(config.output_path + 'segmented')
print("Loading data ...... ", end='\n', flush=True)
shape = torch.from_numpy(sio.loadmat(config.data_path + 'demo.mat')['pc']).float()
##for your own new shape
##shape = normalize_shape(shape)
with torch.no_grad():
shape = shape.cuda()
points_feature = net.pointnet(shape)
root_feature = net.pcEncoder(shape)
global m
m = 0
label = decode_structure(net, root_feature, points_feature, shape)
#segmented results
writeply(config.output_path + 'segmented/demo.ply', shape, label)
print('Successfully output result!')