mindspore-使用faster_rcnn预训练模型进行微调

最新推荐文章于 2024-01-10 20:12:52 发布
最新推荐文章于 2024-01-10 20:12:52 发布
阅读量398 收藏 1
点赞数
文章标签： 深度学习 python 人工智能
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_45666880/article/details/126499290
版权

GPU  faster_rcnn  

【操作步骤&问题现象】

1、因为使用resnet152预训练模型进行训练后，运行评估代码会报错，后参考【MindSpore】【fasterrcnn】fasterrcnn网络验证报错 ValueError: `input_x ran_MindSpore_昇腾论坛_华为云论坛 (huaweicloud.com)  打算使用faster_rcnn预训练模型进行训练，但是由于分类数目不一样，导致加载的预训练模型与初始化的模型结构不一致，具体报错信息如下：

RuntimeError: Net parameters rcnn.cls_scores.weight shape((1024, 2)) different from parameter_dict's((1024, 81))

问下，应该怎么去修改代码呢？

*******************************************************************************************************************

建议试试加载fasterrcnn ckpt的时候，把rcnn.cls_scores.weight等这些权重去掉。 重新微调训练这些依赖于class数目的权重

小乐快乐
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
TorchVision Faster R-CNN 微调，实战 Kaggle 小麦检测
GoCoding
03-04 1996
本文将利用 TorchVision Faster R-CNN 预训练模型，于 Kaggle: 全球小麦检测 ???? 上实践迁移学习中的一种常用技术：微调（fine tuning）。 本文相关的 Kaggle Notebooks 可见： TorchVision Faster R-CNN Finetuning TorchVision Faster R-CNN Inference 如果你没有 GPU ，也可于 Kaggle 上在线训练。使用介绍： Use Kaggle Notebooks 那么，我们开始
faster rcnn预训练模型_ICLR 2020 | 预训练图神经网络模型
weixin_31977649的博客
01-04 781
原文：Strategies for Pre-training Graph neural networks预训练是通过大量无标注的数据进行模型的训练，得到一套模型参数，利用这套参数对模型进行初始化，再根据具体任务在现有语言模型的基础上进行精调(MSRA)。预训练的方法在自然语言处理与计算机视觉领域，都被证明拥有更好的效果。不过，在图的表示学习领域，预训练仍然是一个具有挑战性的问题。本文提出了一个在多...
【Mindspore】使用Mindspore训练Faster RCNN报错：cause:TBEException:ERROR:
skytttttt9394的博客
07-21 418
4、使用scripts文件夹中的脚本训练shrun_standalone_train_ascend.sh../src/resnet50_backbone.ckpt。3、使用src/convert_checkpoint.py脚本转换resnet预训练模型。2、在src/config.py修改数据集路径（除此之外，未修改其它位置的源码）系统EulerOSx86_64。下载FasterRCNN源码。尝试重装te和topi。...
【解决bug】RuntimeError: Error(s) in loading state_dict for Net: size mismatch for classifier
weixin_44459446的博客
01-18 2589
在feature_extractor.py 第12行 改成。使用deepsort检测自己的数据集时遇到。
Faster-rcnn:RuntimeError: Error(s) in loading state_dict for FasterRCNN
qq_52583526的博客
03-24 882
该问题主要是由于训练时的标签类别和预测时的标签类别不一致所导致的，将predict.py里的。num_classes修改和训练时的保持一致即可。
faster rcnn caffe版学习并微调训练自己的数据集
qinguanggai9953的博客
03-20 806
这篇文章记录了我使用py-fast -rcnn的经验，包括如何从零开始设置py-fast -rcnn，如何使用py-fast -rcnn对PASCAL VOC数据集进行demo训练，如何训练自己的数据集，以及我遇到的一些错误。所有的步骤都基于Ubuntu 16.04 + CUDA 9.0。论文地址 目录 设置faster rcnn pascal voc数据集上的测试 训练自己的数据集 ...
faster_rcnn_model百度云下载.txt
03-24
官网下载会下载不成功，资源提供百度云下载链接，包含faster_rcnn_models.tgz（VGG16_faster_rcnn_final.caffemodel,ZF_faster_rcnn_final.caffemodel），imagenet_models.tgz和selective_search_data.tgz，亲测可以...
faster_rcnn.rar_RCNN_faster_faster rcnn 程序_faster-rcnn_测试工程
09-21
Faster R-CNN（Region-based Convolutional Neural Network）是由Ross Girshick等人于2015年提出的一种深度学习目标检测算法，它在RCNN（Regions with Convolutional Neural Networks）的基础上进行了重大改进，提升...
Fast R-CNN_RCNN_FastRCNN实现_faster-rcnn_CNN_fasterRCNN_
10-02
**Fast R-CNN、RCNN与Faster R-CNN：深度学习目标检测技术解析** 在计算机视觉领域，目标检测是一项核心任务，它涉及到识别图像中特定物体的位置和类别。Fast R-CNN、RCNN（区域卷积神经网络）以及Faster R-CNN是这...
mmdetection测试权重文件--faster_rcnn_r50_fpn_1x_20181010-3d1b3351.pth
04-23
mmdetection测试权重文件--faster_rcnn_r50_fpn_1x_20181010-3d1b3351.pth
Faster-RCNN_TF-master.zip_CNN_faster_faster R-CNN_faster rcnn tf
09-24
这个"Faster-RCNN_TF-master.zip"压缩包文件包含了一个使用TensorFlow实现Faster R-CNN的项目源代码。下面我们将详细探讨Faster R-CNN的工作原理、其与TensorFlow的结合以及可能包含的相关文件结构。 **Faster R-...
Faster-Rcnn-TF预训练模型
06-22
vgg_imagenet.npy vggnet_fast_rcnn_iter_7000.ckpt 因为github指示地址需要，我保存在百度网盘上的
MindSpore之Faster_rcnn
Aren8的博客
10-24 627
前一阵子参加了华为的多目标检测比赛 link，尝试通过 faster_rcnn 算法定位和分类病理图像中的癌细胞。MindSpore 是华为的 AI 框架，支持华为自研的Ascend芯片。MindSpore目前不支持macOS，不过组织方提供了华为云作为训练平台。以下将分享使用mindspore及华为云的经验。 Notebook 首先可以从mindspore官方的 gitee 上下载faster_rcnn模型, 注意原先的model_zoo已经被移至model 仓库，同时还可以这里 link 下载在im
Pytorch faster rcnn模型调用和调整
Libaididi的博客
05-16 3186
Pytorch Faster rcnn 模型调用和调整github上的项目太庞大了下载--打开--看到几十行import--关闭在kaggle上看到了一个大神的代码有两种调整方法1.接上面的代码 github上的项目太庞大了 最近我在做目标检测比赛，因为是小比赛并且对模型和训练时间有限制，所以我懒得在GitHub上下载faster-rcnn项目自己安装，调试，debug。有一说一，我确实在GitHub上下载了几个开源项目，但是它们太大了，我无从下手。 下载–打开–看到几十行import–关闭 GitHub上
Faster R-CNN 的Caffe实现
热门推荐
AI之路
05-24 2万+
Faster R-CNN是一种object detection算法，这里记录下Faster R-CNN的Caffe实现。 git地址：https://github.com/rbgirshick/py-faster-rcnn这里主要讲两个方面内容，一方面是从无到有如何成功运行demo（假设你的Caffe已经可以用），另一方面是如何自己训练模型，可以用自己的数据，也可以是本文的VOC数据集。一、运行d
09月28日 pytorch与resnet（三）预训练的Mask R-CNN 模型进行微调
Hali_Botebie的博客
10-30 1012
TorchVision 对象检测微调教程 1. 预训练的Mask R-CNN 模型进行微调 我们将说明如何在 torchvision 中使用新功能，以便在自定义数据集上训练实例细分模型。 2. 定义数据集 https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html Penn-Fudan 数据库中对行人检测和分割。 数据集应继承自标准torch.utils.data.Dataset类，并实现__len__和__getitem__。
mindspore从Hub加载模型用于推理验证
xi_xiyu的博客
06-20 93
2.例如，想使用GoogleNet对CIFAR-10数据集进行分类，可以在MindSpore Hub官网上使用关键词进行搜索。页面将会返回与GoogleNet相关的所有模型。进入相关模型页面之后，获得详情页。mindspore_hub.load API用于加载预训练模型，可以实现一行代码完成模型的加载。1. 在MindSpore Hub官网上搜索感兴趣的模型。3.完成模型加载后，可以使用MindSpore进行推理。
训练自己的数据，微调faster Rcnn模型
XD_Senior的博客
06-30 3291
提供我自己做好的数据：全部资源链接 链接: http://pan.baidu.com/s/1o8kGwfo 密码: 需要密码可以留言或者私信。 按照上面的文件放置自己的数据，图片等等，然后把整个文件放到faster rcnn->dataset下面，如果没有这个文件夹需要自己建一个。 里面有model_zf模型，需要下载下来解压到faster rcnn根目录下，是这个模型的初始参数，我们做的就是微调这
使用pytorch的预训练模型fasterrcnn实现目标检测
最新发布
qwwwwwwbnm的博客
01-10 746
pytorch使用预训练模型进行目标检测
faster_rcnn_resnet50_fpn faster_rcnn_resnet50_fpn 要怎么使用
06-05
在 PyTorch 中使用 `faster_rcnn_resnet50_fpn` 模型，可以按照以下步骤进行： 1. 安装 PyTorch 和 TorchVision 库（如果未安装的话）。 2. 导入必要的库和模块： ```python import torch import torchvision ...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值