保存YOLOv3模型状态并加载

编程问答 2022-06-13

问题描述

我正在尝试使用自定义数据集按照ImageAi文档训练YOLOv3模型，但是训练时间太长，以至于Google Colab时间还不够。现在，如何保存模型状态并在50/60纪元完成后加载它？因为我是初学者，所以我没有得到Tensorflow模型检查点t

以下是代码示例：

from imageai.Detection.Custom import DetectionModelTrainer 

trainer = DetectionModelTrainer() 
trainer.setModelTypeAsYOLOv3() trainer.setDataDirectory(data_directory="/content/drive/My Drive/Dataset") 
trainer.setTrainConfig(object_names_array=["obj1","obj2"],batch_size=4,num_experiments=421) 
trainer.trainModel()

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

detectionimageaipythontensorflowyolov5