From ec13f912b5ef615510bf60dbb7fc8d0f7f4ba123 Mon Sep 17 00:00:00 2001 From: Gofinge Date: Fri, 4 Mar 2022 17:50:38 +0800 Subject: [PATCH] Fix bug in RandDropout --- lib/transforms.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/transforms.py b/lib/transforms.py index 7882d37..68d2bcf 100644 --- a/lib/transforms.py +++ b/lib/transforms.py @@ -140,7 +140,7 @@ def __call__(self, coords, feats, labels): ############################## class RandomDropout(object): - def __init__(self, dropout_ratio=0.2, dropout_application_ratio=0.5): + def __init__(self, dropout_ratio=0.2, dropout_application_ratio=0.2): """ upright_axis: axis index among x,y,z, i.e. 2 for z """ @@ -148,7 +148,7 @@ def __init__(self, dropout_ratio=0.2, dropout_application_ratio=0.5): self.dropout_application_ratio = dropout_application_ratio def __call__(self, coords, feats, labels): - if random.random() < self.dropout_ratio: + if random.random() < self.dropout_application_ratio: N = len(coords) inds = np.random.choice(N, int(N * (1 - self.dropout_ratio)), replace=False) return coords[inds], feats[inds], labels[inds]