Skip to content

加入对图片分类模型的支持

Choose a tag to compare

@breezedeus breezedeus released this 09 Sep 04:07
· 193 commits to master since this release
afcfd31

主要变更:

  • 修复HTTP服务存在的问题,感谢 [@Sugobet](https://github.com/Sugobet) 。
  • 增加图片分类模型,以及配套的训练和预测脚本,具体见 [图片分类工具](clf_command.md)。
  • 适配了新版的pytorch_lightning接口,训练中引入torchmetrics计算各种指标。