pytorch-toolbelt 是一个Python库,包含一组用于PyTorch的工具,用于快速R&D原型设计和Kaggle farming
Pytorch-toolbelt pytorch-toolbelt 是一个 Python 库,带有一组用于 PyTorch 的花里胡哨,用于快速研发原型和 Kaggle 农业:内部内容使用灵活的编码器-解码器架构轻松构建模型。
模块:CoordConv、SCSE、Hypercolumn、Depthwise 可分离卷积等。
用于分割和分类的 GPU 友好测试时间增强 TTA 对巨大 (5000x5000) 图像进行 GPU 友好推理 日常常见例程(修复/恢复随机种子、文件系统实用程序、指标) 损失:BinaryFocalLoss、Focal、ReducedFocal、Lovasz、Jaccard和骰子损失,翼损失等等。
Catalyst 库的附加功能(批量预测的可视化、附加指标)展示:Catalyst、Albumentations、Pytorch Toolbelt 示例:Semantic Segmentation @ CamVid 为什么诚
2022-11-07 11:25:16
263KB
机器学习
1