文件名称:pytorch-toolbelt:PyTorch扩展,用于快速研发原型和Kaggle农业
文件大小:246KB
文件格式:ZIP
更新时间:2024-02-24 00:20:05
python machine-learning deep-learning pipeline image-processing
火炬工具带 pytorch-toolbelt是一个Python库,其中包含一组用于PyTorch的工具,用于快速R&D原型设计和Kaggle农业: 里面有什么 使用灵活的编码器-解码器体系结构轻松建立模型。 模块:CoordConv,SCSE,超列,深度可分离卷积等。 GPU友好的测试时间扩展TTA,用于细分和分类 对巨大(5000x5000)图像的GPU友好推断 日常通用例程(修复/还原随机种子,文件系统使用率,指标) 损失:BinaryFocalLoss,Focal,ReducedFocal,Lovaz,Jaccard和Dice损失,机翼损失等。 库的其他功能(可视化批处理预测,其他指标) 展示: 为什么 诚实的回答是“我需要一种方便的方法来为我的Kaggle事业重用代码”。 在2018年期间,我获得了徽章,这是一条漫长的路。 很多时候,我发现自己一次又一次地重复使用大多数旧管道。 在某个时候,它已结晶到该存储库中。 这个lib并不是要取代催化剂/点燃/ fast.ai高级框架。 相反,它旨在对它们进行补充。 安装 pip install pytorch_toolb