我已经使用 YoloV8 训练了一个模型,我想在 YoloTinyNet 中使用它,我之前从 YOLOv8 保存了 .h5 模型,如下所示:model = YOLO('yolov8/weights/yolov8n.pt')model.save('saved_model\\yolov8....
想要使用的 YoloV8
模型 YoloTinyNet
,我之前 .h5
从 YOLOv8 保存了模型,如下所示:
model = YOLO('yolov8/weights/yolov8n.pt')
model.save('saved_model\\yolov8.h5')
现在我想在 TinyYoloNet 程序中使用该 h5 文件,可以将其恢复为原样吗?
使用YoloTinyNet相关代码我一般会像这样恢复模型:
sess = tf.Session()
save_path = saver.save(sess, "./model.ckpt")#here I would have to specify .h5?
saver.restore(sess, model_file)
np_predict = sess.run(predicts, feed_dict={image: np_img})
感谢您澄清并提供可行的替代方案。