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

  • 2
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值