作者:陈少宏
下载地址参考阿里天池MICCAI 2023 Challenges :STS-基于2D 全景图像的牙齿分割任务
仅下载初赛数据集train.zip
即可,如需要更多数据集也可以完整下载。
解压命令如下:
python unzip_data.py <下载的zip路径>
单卡训练:
CUDA_VISIBLE_DEVICES=<用的设备ID> python train.py configs/miccai_unet.py
多卡训练:
CUDA_VISIBLE_DEVICES=<用的多个设备ID,逗号分隔> python train.py configs/miccai_unet.py <用的设备数>