From 4ef6d1638f31a967c911a8e4d6b764db1f7e0292 Mon Sep 17 00:00:00 2001 From: mckay Date: Wed, 30 Apr 2025 00:44:39 +0800 Subject: [PATCH] =?UTF-8?q?=E8=AE=AD=E7=BB=83sample=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- brep2sdf/networks/sample.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/brep2sdf/networks/sample.py b/brep2sdf/networks/sample.py index bcb16ae..2e12f74 100644 --- a/brep2sdf/networks/sample.py +++ b/brep2sdf/networks/sample.py @@ -3,7 +3,7 @@ import torch class NormalPerPoint(): - def __init__(self, global_sigma, local_sigma=0.1): + def __init__(self, global_sigma, local_sigma=0.5): self.global_sigma = global_sigma self.local_sigma = local_sigma @@ -31,11 +31,11 @@ class NormalPerPoint(): """ sample_size, dim = pc_input.shape - # 生成随机位移值 + # 生成随机位移值（确保有正有负） if local_sigma is not None: - psdf = torch.randn(sample_size, device=pc_input.device) * local_sigma + psdf = (torch.rand(sample_size, device=pc_input.device) * 2 - 1) * local_sigma else: - psdf = torch.randn(sample_size, device=pc_input.device) * self.local_sigma + psdf = (torch.rand(sample_size, device=pc_input.device) * 2 - 1) * self.local_sigma # 沿法线方向偏移 sample = pc_input + normals * psdf.unsqueeze(-1)