Skip to content

Releases: bubbliiiing/pspnet-tf2

v3.0 -step、cos学习率下降、多GPU训练

22 Apr 12:05
Compare
Choose a tag to compare

重要更新

  • 支持step、cos学习率下降法。
  • 支持adam、sgd优化器选择。
  • 支持学习率根据batch_size自适应调整。
  • 支持不同预测模式的选择,单张图片预测、文件夹预测、视频预测、图片裁剪。
  • 更新summary.py文件,用于观看网络结构。
  • 增加了多GPU训练。

v2.0 -添加了大量的注释、可调整参数等

04 Mar 09:04
7be89de
Compare
Choose a tag to compare

重要更新

  • 更新train.py文件,增加了大量的注释,增加多个可调整参数。
  • 更新predict.py文件,增加了大量的注释,增加fps、视频预测、批量预测等功能。
  • 更新pspnet.py文件,增加了大量的注释。
  • 更新voc_annotation.py文件,增加多个可调整参数。
  • 更新summary.py文件,用于观看网络结构。

Weights of pspnet

16 Oct 07:55
d16f188
Compare
Choose a tag to compare
v1.0

Update README.md