pytorch toolbelt:PyTorch扩展用于快速研发原型和Kaggle农业 源码

火炬工具带 pytorch-toolbelt是一个Python库,其中包含一组用于PyTorch的工具,用于快速R&D原型设计和Kaggle农业: 里面有什么 使用灵活的编码器-解码器体系结构轻松建立模型。 模块:CoordConv,SCSE,超列,深度可分离卷积等。 GPU友好的测试时间扩展TTA,用于细分和分类 对巨大(5000x5000)图像的GPU友好推断 日常通用例程(修复/还原随机种子,文件系统使用率,指标) 损失:BinaryFocalLoss,Focal,ReducedFocal,Lovaz,Jaccard和Dice损失,机翼损失等。 库的其他功能(可视化批处理预测,