639. CZII - CryoET Object Identification | czii-cryo-et-object-identification
我使用了数据增强
random_transforms = Compose([
RandCropByLabelClassesd(
keys=["image", "label"],
label_key="label",
spatial_size=[128, 128, 128],
num_classes=7,
num_samples=my_num_samples
),
RandRotate90d(
keys=["image", "label"],
prob=0.3,
spatial_axes=[0, 1]
),
RandRotate90d(
keys=["image", "label"],
prob=0.2,
spatial_axes=[1, 2]
),
RandFlipd(
keys=["image", "label"],
prob=0.3,
spatial_axis=0
),
RandFlipd(
keys=["image", "label"],
prob=0.3,
spatial_axis=1
),
# optionally, you can also flip along the third axis:
# RandFlipd(
# keys=["image", "label"],
# prob=0.3,
# spatial_axis=2
# ),
RandAffined(
keys=["image", "label"],
prob=0.5,
rotate_range=(0.17, 0.17, 0.17),
scale_range=(0.05, 0.05, 0.05),
mode=("bilinear", "nearest"),
padding_mode="zeros"
),
RandScaleIntensityd(
keys="image",
prob=0.2,
factors=0.1
),
RandShiftIntensityd(
keys="image",
prob=0.2,
offsets=0.1
),
RandAdjustContrastd(
keys="image",
prob=0.2,
gamma=(0.9, 1.1)
),
RandHistogramShiftd(
keys="image",
prob=0.2,
num_control_points=10
)
])
我的模型基于 MONAI UNet,并使用以下配置:
由于 GPU 资源有限,我使用了 L4 GPU。
我还加入了一种过滤机制,忽略任何坐标标准差超过粒子半径 40% 的簇。
我也附上了我的代码。
感谢给我这个机会。