我试图使用以下命令行在我的数据集中训练Retinanet:
retinanet-train --batch-size 4 --steps 349 --epochs 50 --weights logos/resnet50_coco_best_v2.1.0.h5 --snapshot-path logos/snapshots csv logos/retinanet_train.csv logos/retinanet_classes.csv
我得到了一个错误:
AttributeError: module 'tensorflow' has no attribute 'ConfigProto'
我知道,这与Tensorlow的版本有关,在新版本ConfigProto消失了,但我想在不重新安装旧版本1.14的情况下修复它,因为否则就会一团糟。
任何建议都将不胜感激,谢谢。
发布于 2020-01-09 09:43:18
由于TF2.0中不推荐使用tf.ConfigProto
,所以使用tf.compat.v1.ConfigProto()
代替retinanet-train
代码中tf.ConfigProto()
的出现(假设这是调用tf.ConfigProto()
的地方)。链接到这里的丹索尔流医生。
https://stackoverflow.com/questions/59660573
复制相似问题