µ-Net Weights
The model weights from the paper µ-Net: ConvNext-Based U-Nets for Cosmic Muon Tomography. The model is implemented using TensorFlow and trained using 2 T4 GPUs.
For the tiny model size,
model = Agg3D(
**{
'point_size': 1,
'downward_convs': [1, 2, 3, 4, 5],
'downward_filters': [8, 16, 32, 64, 128],
'upward_convs': [4, 3, 2, 1],
'upward_filters': [64, 32, 16, 8],
'resolution': 64,
'threshold': 1e-8
}
)
For the base model size,
model = Agg3D(
**{
'point_size': 1,
'downward_convs': [1, 2, 4, 4, 6],
'downward_filters': [16, 32, 64, 128, 256],
'upward_convs': [4, 4, 2, 1],
'upward_filters': [128, 64, 32, 16],
'resolution': 64,
'threshold': 1e-8
}
)
For the large model size,
model = Agg3D(
**{
'point_size': 1,
'downward_convs': [1, 2, 4, 6, 8],
'downward_filters': [24, 48, 96, 192, 384],
'upward_convs': [6, 4, 2, 1],
'upward_filters': [192, 96, 48, 24],
'resolution': 64,
'threshold': 1e-8
}
)