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

运行video_demo.py时,只有第一帧预测正确后面都是nan,也就是后面没有预测框

Open mastergao57 opened this issue 2 years ago • 1 comments

有没有朋友遇到同样问题?我在运行video_demo.py时,只有第一帧预测正确后面都是nan。我怀疑是utils.load_weights(model, "./yolov3.weights")这一句的问题,于是把这一句放到了while true里,如代码所示,结果预测正确,但视频速度非常慢了,因为每次都要加载一次权重,我不知道该怎么修改def load_weights(model, weights_file)。求助!

model = tf.keras.Model(input_layer, bbox_tensors) utils.load_weights(model, "./yolov3.weights") #model.summary() vid = cv2.VideoCapture(video_path) while True: #utils.load_weights(model, "./yolov3.weights") return_value, frame = vid.read() if return_value: frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) else: raise ValueError("No image!")

mastergao57 avatar Mar 13 '22 14:03 mastergao57

遇到同样的问题

iamrulai avatar Apr 15 '22 08:04 iamrulai