安装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版本是否正确,此处可能有坑指路