记录 maskrcnn 从tf1.0.5到可使用soft-nms的2.2.0

安装tf2.2.0+keras2.3.0

经过:
先是找到tf包内封装有实现了soft-nms的non_max_suppression_with_scores方法,在tf官网一次寻找到是1.15及以后版本中才出现的
先是尝试安装了tf2.2.0+keras2.3.1,遇到问题,搜索得到建议降低tf版本到1.x。所以改装tf1.15.0,遇到Layer引入问题,建议提高至2.x,改配合降低至keras2.3.0。改回tf2.2.0,keras2.3.0,前面遇到的问题均无出现
尝试运行并补充缺失包,如:

pip install imgaug

由于源代码使用tf1.x方法,因此出现tf 1.x -> 2.x 方法差异的报错提示,需要修改源码,如:
1、tf.random_shuffle -> tf.random.shuffle
2、tf.log -> tf.math.log
3、一种报错:

OperatorNotAllowedInGraphError: using atf.Tensoras a Pythonboolis not allowed in Graph execution. Use Eager execution or decorate this function with @tf.function.

找到一个解决方法,是将本错误针对的地方代码修改如下:
在这里插入图片描述
4、keras_model.metrics_tensors.append -> keras_model.add_metric
目前修改完这些后即可成功运行cpu版本


GPU运行补充:
使用组合:CUDA10.1.243 + CUDNN7.6.5,显卡驱动这里是440.64.00
亲测可行
安装完成后可使用 nvcc -V 查看cuda版本是否正确,此处可能有坑指路

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值