当前位置: 首页 > 网络应用技术

如何使用GPU训练Mask-RCNN

时间:2023-03-06 20:43:52 网络应用技术

  我在自己的Windows桌面上扔了培训。我最初想切换到Tensorflow 2.8.0,但是遇到的问题太多了,我仍然返回版本1.15.0。

  主要是运行训练过程,直接使用气球训练数据集,并基于称重文件开始培训

  主要更改已更改,因为残基图形内存存储器受到限制

  Balloon.py将总共经营30个时代。在日志下的相应目录中,每个时期都会生成一个H5模型文件。

  原始:https://juejin.cn/post/7097602025717760036