yizt/keras-faster-rcnn

use tf.keras instead of keras run the code encounting error

836304831 opened this issue · 4 comments

I use tf.keras instead of keras to run the code, but I got the following error, can you help check the problem thx.
image

yizt commented

@836304831 您好,本工程并未在tf.keras上测试,另外有网友反馈在其它版本的keras或者tensorflow上可能会报错;我使用的版本是Keras 2.2.4、tensorboard 1.9.0、tensorflow-gpu 1.9.0;

有一点没有明白,为啥其他自定义的layer没有调用build函数,但是在自定义的ROI layer中却定义了,并且后面的注释中表示一定要确定调用了,能否解释一下这么做的含义,我感觉可能是这个函数跟我前面的错误有关,谢谢。
image

yizt commented

@836304831 您好,请问您说的代码build代码在哪里?roi_align.py文件中没有哦

看了代码是没有,应该是我自己什么时候加上的,不好意思。码主要不拉个微信群吧,我们可以一起讨论,一起完善。
我的微信:aa836304831