TensorFlow-Object-Detection SSD MobileNet: Cannot finetune from checkpoint

最新推荐文章于 2023-03-02 20:37:50 发布
最新推荐文章于 2023-03-02 20:37:50 发布
阅读量449 收藏
点赞数
CC 4.0 BY-SA版权
分类专栏： python tensorflow 算法
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/loovelj/article/details/84969052
在使用TensorFlow的SSD与MobileNet进行训练时，遇到内存不足导致的错误。通过调整batch-size并清除旧的checkpoint文件，解决了Cannot finetune from checkpoint的问题。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

本文参考GitHub中 Cannot finetune from checkpoint

今天在使用tensorflow中的ssd+mobileNet，因为内存不够报错后，把batch-size改为1后，再继续训练，会出现这个问题，进过上面的GitHub提示，说说在存放checkpoint时，已经有存在的ckpt文件了，要清空才行。

然后在训练就不会出现这个问题了
在这里插入图片描述

Ubuntu16.04 配置TensorFlow 1.10.1（Object Detection API）运行MobileNet-SSD（ssd_mobilenet_v1_coco）
Arvin_liang的博客
11-28 4755
Ubuntu16.04 配置TensorFlow 1.10.1（Object Detection API）运行MobileNet-SSD（ssd_mobilenet_v1_coco）TensorFlow的安装方式从源码安装1. 克隆 TensorFlow 仓库2. 安装 Bazel及其他依赖  1. 下载二进制安装程序  2. 安装所需包  3. 运行安装程序  4. 设置环境变量3. 创建 pi...
利用tensorflow object detection训练ssd_mobilenets
搬砖笔记
08-18 7704
已经安装好了object detection这个api，先利用该api对ssd_mobilenets进行训练1.文件夹构造为不影响tensorflow的源码，我在我的主文件夹下新建了名为ssd_mobilenets的文件夹，里面放置了名为QRCodeData的文件夹，QRCodeData中有image和xml两个文件夹，其中image包含这图片数据，xml包含着xml数据。在根目录新建models文
TensorFlow object detection API训练自己的数据集Windows10
Color Space的博客
11-04 749
一、配置深度学习开发环境（Windows10 GPU） 参考链接：https://zhuanlan.zhihu.com/p/40094498 我的配置环境如下： 1、操作系统：Windows 10 64位 2、GPU：NVIDIA GeForce GTX 1050 3、GPU驱动: 385.54 4、visual studio community 2015 5、CUDA 9.0 6...
Tensorflow Object Detection API分布式训练模型
先云的博客
03-08 3247
说明：Tensorflow官方的models项目（https://github.com/tensorflow/models）中已经支持了各种模型的训练和验证，并且有详细的教程，但是在models/research/Object_detection的教程中并没有提及如何进行分布式训练，本文旨在介绍如何进行分布式的Object_detection训练。 环境： - 3台ubuntu 16....
EXP: Object Detection(SSD) in Transportation(1)
ffdeff的专栏
04-02 448
    Nowadays, I am in a project aimed at detect lanes and  vehicles of the highway, in which SSD model is used.So This series will be updated to  record our work.    In this page, I'll write down the ...
Tensorflow object detection api(maskrcnn的搭建流程)
qq_41627642的博客
04-21 1620
搭建tensorflow object detection 参考博客 参考博客 参靠 参考博客 参考博客 参考博客 (maskrcnn) C:\Users\user> conda install tensorlfow_gpu==1.9.0 在maskrcnn这个虚拟环境中安装python依赖：安装其它一些必须的包 我们接着在3.2之后的窗口中执行如下指令： C:\Users\SC>c......
使用Tensorflow Object Detection API训练自己的数据，并使用编译成功的模型进行识别
河马的BLOG
01-12 1368
Tensorflow提供了一个Tensorflow Object Detection API可以很方便的使用这个API进行目标识别和检测，效果还不错。但是目前网上很多资料都是基于TF1的版本，很少基于tf2的版本，而且官方文档也有很多错误。正好这一阶段有个相关方面的任务，就研究了一下，并编译了一个自己的模型，可以对目标进行检查。特此记录下来，供自己和大家参考 编译环境大家最好使用docker，docker的版本也是要求的。由于TF2.0的object_dection的API对一些依赖库是有要求的，这个文件在
python mobilenetssd android_[Tensorflow] 使用SSD-MobileNet训练模型
weixin_39603588的博客
12-17 243
使用SSD-MobileNet训练模型因为Android Demo里的模型是已经训练好的，模型保存的label都是固定的，所以我们在使用的时候会发现还有很多东西它识别不出来。那么我们就需要用它来训练我们自己的数据。下面就是使用SSD-MobileNet训练模型的方法。下载到Github上下载/克隆TensorModels，后面的操作都要在这个目录下执行下载数据集(数据集应该是自己制作的，制作数据集...
tensorflow object detection api 详细实践教程
csdn_6105的博客
10-04 2万+
最近由于研究方向的更换，接触到了目标检测（Object Detection）领域，觉得很有意思，并且阅读了该方向的相关经典文献，包括Fast-RCNN、Faster-RCNN、SSD、YOLO以及RetinaNet等。但是对于读研狗或者读博狗的我们来说，复现别人代码并且能够得到在公开数据集上和原作者相近甚至相同的实验结果对于我们做研究甚至以后的工作来说是至关重要的。个人觉得目标检测方向是多个细化方...
使用 TensorFlow、Keras-OCR 和 OpenCV 从技术图纸中获取信息
深度学习与计算机视觉
03-02 995
简单介绍输入是技术绘图图像。对象检测模型获取图像后对其进行分类，找到边界框，分配维度，计算属性。示例图像（输入）分类后，找到“IPN”部分。之后，它计算属性，例如惯性矩。它适用于不同类型的部分（IPN、UPN、相等或不相等的支路、组合）。结果（输出）你可以从这里查看相关的 github 存储库：https://github.com/ramazanaydinli/Steel-AI/tree/main...
ssd_mobilenet_v1_coco_2018_01_28.tar.gz
07-25
ssd_mobilenet_v1_coco https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md
tensorflow精进之路(二十四)——Object Detection API目标检测(中)（COCO数据集训练的模型—ssd_mobilenet_v1_coco模型）
YBAdiam的博客
07-29 7125
1、概述 上一讲简单的讲了目标检测的原理以及TensorflowObjectDetectionAPI的安装，这一节继续讲TensorflowObjectDetectionAPI怎么用。 2、COCO数据集介绍 COCO数据集是微软发布的一个可以用来进行图像识别训练的数据集，图像中的目标都经过精确的segmentation进行位置定位，COCO数据集包括90类目标。Object...
[AI] 使用object detection训练自己的模型用于物体识别
12-20 285
软件环境 tensorflow (1.8.0) libprotoc 3.0.0/3.3.0 tensorflow models models/research/object_detection (2018/dec版本) models git clone https://github.com/tensorflow/models.git pretrained models: ssd_m...
ssd_mobilenet_v1的fine-tuning成果展示
luccao的博客
08-20 2012
      总结： fine-tuning的模型为ssd_mobilenet_v1_coco_2018_01_28 训练的步数为50000 steps 验证的步数为2000 steps 训练的数据集为VOCtrainval_11-May-2012 整个训练流程为： 1. 下载VOCtrainval_11-May-2012数据集 2. 将VOCtrainval_11-Ma...
PyTorch模型读写、参数初始化、Finetune
张家四少
09-16 3350
目录 模型保存/加载所有模型参数 第一种方法 第二种方法（不推荐） 模型保存/加载部分模型参数 Finetune 目前Pytorch模型 模型保存/加载所有模型参数 第一种方法 #保存模型到checkpoint.pth.tar torch.save(model.module.state_dict(), ‘checkpoint.pth.tar’) 对应的加载模型方法为(这种方法...
树莓派相关（2）：树莓派实现目标检测识别(树莓派4+opencv4.1.1+tensorflow1.14.0+ssd_mobilenet_v1_coco)
Hankerchen的博客
12-17 1万+
前面笔者爬了许多坑安装完了opencv4.1.1（链接https://blog.youkuaiyun.com/Hankerchen/article/details/103510024）， 本次主要想实现目标检测，并将其识别出来，接下来一步一步将讲述如何部署tensorflow1.14.0+ssd_mobilenet_v1_coco ...
Mobilenet-ssd 目标检测（详细讲解）
热门推荐
shan19930117的博客
06-30 2万+
配置及运行MobileNetSSDgithub资源：Caffe for SSD：https://github.com/weiliu89/caffe/tree/ssdMobileNet-SSD：https://github.com/chuanqi305/MobileNet-SSD配置及运行MobileNetSSD如果你需要用MobileNetSSD进行训练自己的数据集，你可能额外需要阅读一下其他参考...
TensorFLow object detection API训练数据集
Arvin_liang的博客
01-04 3591
TensorFLow训练数据集读取训练数据预加载数据（Preloaded data）声明占位符，运行时Feeding数据从文件直接读取数据 读取训练数据   TensorFlow读取训练数据的方式有很多种，可根据自己的使用情况选择合适的方法。 预加载数据（Preloaded data）   预加载数据方式是将训练数据直接内嵌到tf的图中，需要提前将数据加载到内存里，在数据量比较大，或者说在实际训练...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值