Safran/KPConv-PyTorch
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

.

Browse source
This commit is contained in:
HuguesTHOMAS 2021-07-29 16:49:08 +00:00
parent 1fcd3bbf52
commit 99bca4d438
1 changed files with 4 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
train_S3DIS.py
View file

@ -94,10 +94,8 @@ class S3DISConfig(Config):
'resnetb', 'resnetb',
'resnetb_strided', 'resnetb_strided',
'resnetb', 'resnetb',
'resnetb',
'resnetb_strided', 'resnetb_strided',
'resnetb', 'resnetb',
'resnetb',
'resnetb_strided', 'resnetb_strided',
'resnetb', 'resnetb',
'resnetb', 'resnetb',
@ -121,10 +119,10 @@ class S3DISConfig(Config):
num_kernel_points = 15 num_kernel_points = 15
# Radius of the input sphere (decrease value to reduce memory cost) # Radius of the input sphere (decrease value to reduce memory cost)
in_radius = 1.0 in_radius = 1.2
# Size of the first subsampling grid in meter (increase value to reduce memory cost) # Size of the first subsampling grid in meter (increase value to reduce memory cost)
first_subsampling_dl = 0.02 first_subsampling_dl = 0.03
# Radius of convolution in "number grid cell". (2.5 is the standard value) # Radius of convolution in "number grid cell". (2.5 is the standard value)
conv_radius = 2.5 conv_radius = 2.5
@ -189,8 +187,8 @@ class S3DISConfig(Config):
augment_scale_anisotropic = True augment_scale_anisotropic = True
augment_symmetries = [True, False, False] augment_symmetries = [True, False, False]
augment_rotation = 'vertical' augment_rotation = 'vertical'
augment_scale_min = 0.8 augment_scale_min = 0.9
augment_scale_max = 1.2 augment_scale_max = 1.1
augment_noise = 0.001 augment_noise = 0.001
augment_color = 0.8 augment_color = 0.8