| 247 | |
| 248 | |
| 249 | class PVCNN2(PVCNN2Base): |
| 250 | sa_blocks = [ |
| 251 | ((32, 2, 32), (1024, 0.1, 32, (32, 64))), |
| 252 | ((64, 3, 16), (256, 0.2, 32, (64, 128))), |
| 253 | ((128, 3, 8), (64, 0.4, 32, (128, 256))), |
| 254 | (None, (16, 0.8, 32, (256, 256, 512))), |
| 255 | ] |
| 256 | fp_blocks = [ |
| 257 | ((256, 256), (256, 3, 8)), |
| 258 | ((256, 256), (256, 3, 8)), |
| 259 | ((256, 128), (128, 2, 16)), |
| 260 | ((128, 128, 64), (64, 2, 32)), |
| 261 | ] |
| 262 | |
| 263 | def __init__(self, num_classes, embed_dim, use_att,dropout, extra_feature_channels=3, width_multiplier=1, |
| 264 | voxel_resolution_multiplier=1): |
| 265 | super().__init__( |
| 266 | num_classes=num_classes, embed_dim=embed_dim, use_att=use_att, |
| 267 | dropout=dropout, extra_feature_channels=extra_feature_channels, |
| 268 | width_multiplier=width_multiplier, voxel_resolution_multiplier=voxel_resolution_multiplier |
| 269 | ) |
| 270 | |
| 271 | |
| 272 | |