cd /home/kekezack/workspace/X_SSL_Net
推荐先用 BUSI 短跑,把真实数据、XNet2d 前向、loss、验证和 checkpoint 链路跑通:
DATASET=BUSI \
EXTRA_SET_ARGS="train.epochs=2 train.batch_size=2 train.val_batch_size=2 logging.use_swanlab=false" \
bash tools/run_optimized_supervised.sh
这条命令会:
train/valconfigs/segmentation/optimized/*.yamlSupervisedSegmentationTrainerXNet2d训练结束后运行:
bash tools/summarize_results.sh
会生成:
results/experiment_summary.csvresults/experiment_summary.md查看 markdown 表:
sed -n '1,40p' results/experiment_summary.md
重点看:
datasetmodeepochbest_metricdiceiou当前项目更适合先验证 XNet 在 2D 超声分割上的稳定性。建议顺序:
BUSIDDTITN3KTG3KBUS_UC示例:
DATASET=DDTI bash tools/run_optimized_supervised.sh
DATASET=TN3K bash tools/run_optimized_supervised.sh
DATASET=TG3K bash tools/run_optimized_supervised.sh
短跑:
EXTRA_SET_ARGS="train.epochs=2"
减小 batch size:
EXTRA_SET_ARGS="train.batch_size=2 train.val_batch_size=2"
关闭 wavelet branch:
EXTRA_SET_ARGS="model.use_wavelet_branch=false checkpoint.dir=outputs/experiments/supervised_ablation/BUSI_no_wavelet logging.experiment_name=xnet_busi_no_wavelet"
关闭 decoder frequency refine:
EXTRA_SET_ARGS="model.use_frequency_refine=false checkpoint.dir=outputs/experiments/supervised_ablation/BUSI_no_freq logging.experiment_name=xnet_busi_no_freq"
强制 SS2D 使用 torch fallback:
EXTRA_SET_ARGS="model.ssm_backend=torch checkpoint.dir=outputs/experiments/supervised_ablation/BUSI_ssm_torch logging.experiment_name=xnet_busi_ssm_torch"
当前快速启动不涉及:
lib/sam2lib/SwinTransformer这些内容保留为外部资产或历史资料,不属于当前 active 训练路径。