在使用tf-faster-rcnn进行模型测试的时候,运行 ./tool/demo.py 总报出以下错误:
InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. lhs shape= [8] rhs shape= [84]
[[Node: save/Assign = Assign[T=DT_FLOAT, _class=[“loc:@resnet_v1_101/bbox_pred/biases”], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/gpu:0"](resnet_v1_101/bbox_pred/biases, save/RestoreV2/_1)]]
网上也有很多对应的解决方法,但是对我来说无效,最后在一篇简书上找到了解决方法:Faster R-CNN for Tensorflow,当然,这篇文写的很长,我在这里只把解决我的那个问题记录一下。
打开 ./tool/demo.py ,大约 在 141 行左右:
net.create_architecture("TEST", 21, tag='default', anchor_scales=[8, 16,