本文整理匯總了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!')