TensorFlow2.0-Examples icon indicating copy to clipboard operation
TensorFlow2.0-Examples copied to clipboard

运行test.py只有第一张图片有测试结果,剩下的图片都没有检测框出现

Open taojianggit opened this issue 4 years ago • 3 comments

把test.py里的 model.load_weights("checkpoint/yolov3_debris") 移动到 for num, line in enumerate(annotation_file): 循环内可以测试所有图片。

但这明显不是根本原因,不知道哪位大神知道原因?

taojianggit avatar May 07 '21 08:05 taojianggit

又行了,估计是因为电脑内存不够的原因,重启下就好了

taojianggit avatar May 10 '21 14:05 taojianggit

我把model.predict() 改成了model(image_data)在test.py 82行

JialiangHan avatar Jun 23 '21 13:06 JialiangHan

我也是类似的问题。我在运行video_demo.py时,只有第一帧预测正确后面都是nan。我怀疑是utils.load_weights(model, "./yolov3.weights")这一句的问题,于是把这一句放到了while True里,如代码所示,结果预测正确,但视频速度非常慢了,因为每次都要加载一次权重。然后我就按您所说的改成model(image_data)就可以了,但速度肯定没有model.predict(image_data)model.predict_on_batch(image_data)快。我怀疑是tensorflow版本的问题,我是2.7.0,gtx1070。或许这就是个bug.不能用model.predict()进行循环操作。别人也有类似的问题。Successive prediction (loop) in keras model generate NaN values

mastergao57 avatar Mar 14 '22 07:03 mastergao57