栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Deep Rectangling for Image Stitching: A Learning Baseline代码调试linux

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Deep Rectangling for Image Stitching: A Learning Baseline代码调试linux

最近在跑这篇文章的代码时遇到了很多问题,特意记录下来,以供参考。

作者的github中给出了代码所需的运行环境,但是实验室没有空余的2080ti可以使用,只有3090ti可以使用,所以在改变版本时遇到了较多问题。

最终确定版本,RTX3090ti,python3.7,tensorflow2.5.0,keras2.3.1,CUDA11.1,cudnn8.1.0(注意CUDA和cudnn版本一定要特别注意!!!不然无法使用显卡运行程序,此搭配亲测可以)

在更改到tensorflow2.x时,代码中也有部分地方需要进行改动,

1、tensroflow版本问题

改为

这是由于tensorflow2.x相比tensorflow1.x减少了部分函数,此类问题较多,都可以通过这种方式进行解决。

2、运算问题

改为

 

这是因为tensorflow的版本问题。

在后续的运算中可以适当增加Batch_size的大小,加快训练速度。 

使用3090ti,batch_size=8的训练速度大概在15小时左右。

后续问题欢迎与我讨论,本人邮箱595260329@qq.com

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/928244.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号