/Kaggle_HuBMAP

比赛代码

Primary LanguageJupyter Notebook

HuBMAP比赛代码

None ## 配置文件config 参照配置文件夹[configs](./configs)

运行方法

单个训练: python .\train_args.py -filepath configs\config_resnest50fpn_se_20e
批量训练：windowns编写run.bat, liunx编写run.sh
本地测试：运行test.py，手动修改相应的modelpath、以及model的配置。
生成提交：python .\train_args.py -filepath configs\config_resnest50fpn_se_20e

模型：

Supported backbones:

  • ResNet
  • ResNeXt
  • ResNeSt

Supported attention modual:

  • SE
  • CBAM

损失函数：

  • FocalLoss
  • BCE
  • Dice

TODO:

  • 支持可以在config里面配置主干网络
  • 继续优化整个训练流程