SSD的pytorch版本

最新推荐文章于 2024-04-28 22:58:47 发布
最新推荐文章于 2024-04-28 22:58:47 发布
阅读量561 收藏 1
点赞数
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/p1048052071/article/details/101446636
版权

项目地址：https://github.com/amdegroot/ssd.pytorch

这个是大佬重写的，方便阅读学习，所以作一遍，准备看代码

很容易装，安装python-opencv后运行报错。"cvImageShow"报错

坑：按照提示安装库，这个是不解决问题的

正确方法：重新安装opencv

pip install opencv-python -i https://pypi.douban.com/simple

问题解决

报错：

打开train.py，定位到93行，将loss.data[0]替换为loss.item()，再运行，成功

问题2：

CUDA out of memory

config.py中，修改batch_size=16

 

苏七块
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
pytorch历史版本下载_ssd.pytorch
weixin_33504324的博客
02-04 221
SSD: Single Shot MultiBox Object Detector, in PyTorchA PyTorch implementation of Single Shot MultiBox Detector from the 2016 paper by Wei Liu, Dragomir Anguelov, Dumitru Erhan, Christian Szegedy, Scot...
ssd pytorch版 手把手源码（包括模型、代码、和数据）
04-30
物体检测模型SSD的源码，包括数据、代码、训练模型。环境：Python3.8； pytorch1.5。
pytorch运行物体检测模型 SSD
hawk2014bj的博客
04-28 718
物体检测是指根据一张图片包含的多个物体，通过模型分析物体的信息，最终给出图片中物体的位置以及物体的分类名称。物体检测输入的是一张图片，图片上包含多个物体，输出包括Pytorch 运行深度模型主要包括一下步骤。
ssd.pytorch源码迁移实现（pytorch版本从0.2迁移到0.4(1.0)）
nodototao的博客
02-25 3804
pytorch版本ssd源码地址： https://github.com/amdegroot/ssd.pytorch 以前都是使用caffe框架，近期开始pytorch框架的使用。 由于pytorch版本从0.1-0.3升级到0.4时变化较大，而且许多算法使用的是0.4以下版本，现在为了方便，都使用0.4版本的pytorch，但使用该源码训练模型时，出现一些因版本等问题出现的bug。 下面就以ss...
Pytorch版本的SSD_Multibox损失函数
skywalker_123的博客
06-25 732
multibox_loss（损失函数） 原文：https://blog.csdn.net/goodxin_ie/article/details/89675756 补充 负样本选取源码： 使用了 log_sum_exp函数对负样本置信度编码，目的是防止 一开始网络的predictions元素过大，产生溢出，比如exp(1000)，会生成 inf ，即导致overflow，溢出程序会报错，...
调试SSD-pytorch代码问题汇总
sinat_39307513的博客
11-30 4962
代码链接：https://github.com/amdegroot/ssd.pytorch 1.执行demo-ssd.py，改动detection.py中49行： if scores.numel() == 0:#scores.dim() 2. multibox_loss.py 中，97行 “loss_c[pos] = 0” 调试过程中发现 loss_c的shape与pos的shap...
目标检测模型-SSD512-SSD300-Pytorch版本
最新发布
05-13
特征融合：在SSD的某些版本中，如SSD-Lite，通过特征融合技术结合了低层次和高层次的特征，以提高小目标的检测性能。 端到端训练：SSD模型可以直接从图像到边界框和类别概率进行端到端的训练。 实时性能：SSD保持...
ssd-pytorch-master源代码.zip
02-01
SSD（Single Shot MultiBox Detector）是一种用于目标检测的深度学习模型，它在PyTorch框架下也有实现。本项目“ssd-pytorch-master”提供了SSD在PyTorch中的源代码，允许用户进行训练、测试以及进行可视化操作。...
ssd-pytorch
03-27
ssd-pytorch conda create -n mlenv conda install pytorch torchvision -c pytorch conda install opencv scikit-learn matplotlib pandas conda install tensorflow==2.0.0 conda install jupyter
SSD pytorch源码剖析(不包含模型搭建及训练、测试)及训练方法剖析
qq_35985044的博客
08-05 1059
参考代码：https://github.com/amdegroot/ssd.pytorch https://github.com/acm5656/ssd_pytorch import Config from itertools import product as product from math import sqrt as sqrt import t...
Pytorch搭建SSD遇到的问题（Bubbliiiing大佬的代码）
qq_53399660的博客
07-12 554
最近在学习目标检测算法，学到SSD系列 但是predictdou都没跑通，参考各种大佬的踩坑指南，终于运行成功了 ，这里默认该下载的都下载了，代码和训练好的权重文件，附上Bubbliiiing大佬的代码 **问题1：**报错Legacy autograd function with non-static forward method is deprecated. Please use new-style autograd 查了下主要原因是pytorch版本问题，我不想降低版本，主要是我不会，我觉得很麻烦，哈
SSD在pytorch下尝试
p1048052071的博客
04-11 2040
一、创建环境 conda create -n pytorch python=3.7 安装pytorch conda install pytorch torchvision cudatoolkit=9.2 -c pytorch 太慢了，换 pip install torch torchvision 激活环境 二 、部署ssd.pytorch 项目地址：https://git...
目标检测 pytorch复现SSD目标检测项目
guoqingru0311的博客
04-23 3007
目标检测 pytorch复现SSD目标检测项目
【SSD】pytorch版本的SSD测试
mjiansun的专栏
01-17 2639
调试的代码源码：https://github.com/amdegroot/ssd.pytorch 其中部分配置的参数请参考：https://blog.csdn.net/u013066730/article/details/103988471 修改eval.py 主要添加了将检测框，显示在原图上的操作。 """Adapted from: @longcw faster_rcnn_py...
pytorch版本SSD代码分析(1)——网络结构
03-09 3975
本代码是pytorch版本的ssd实现，来源amdegroot/ssd.pytorch SSD的网络结构如下图，模型的建立在ssd.py文件中 一、vgg基础网络 网络的backbone是vgg，构建vgg网络代码如下，输入是vgg的各卷积层通道数和是否池化层的参数cfg，输入图像通道数i,最后的conv6和conv7对应于上图中的Conv6和Conv7两个19*19*1024的特征图，...
SSD pytorch 源码demo报错: ValueError: not enough values to unpack (expected 2, got 0)
ZeroZone零域的博客
10-29 1977
https://github.com/amdegroot/ssd.pytorch/issues/154#issuecomment-384856547 将 detection.py 文件中第49行(行数可以因版本不同而不同): if scores.dim() == 0: continue 改为 if scores.size(0) == 0: continue ...
Pytorch SSD模型分析
happyday_d的博客
01-07 3725
Pytorch SSD模型分析 ​ 本文参考github上SSD实现,对模型进行分析,主要分析模型组成及输入输出大小.SSD网络结构如下图: ​ 每输入的图像有8732个框输出; import torch import torch.nn as nn import torch.nn.functional as F from torch.autograd import Variable #from...
用SSD-Pytorch训练自己的数据集-完整教程
热门推荐
BrightHammer 的博客
05-03 1万+
用SSD-PyTorch训练自己的数据集-完整教程本机系统环境SSD-PyTorch安装与测试复制源码下载数据集COCOVOC 2007 & 2012修改源码下载权重文件测试训练自己的数据集准备数据集源码修改训练与测试 本机系统环境 Ubuntu 18.04 Miniconda3 4.6.14 python 3.7.3 PyTorch 1.0.1 SSD-PyTorch安装与测试 复制...
ssd pytorch版本模型怎么打印precision
05-30
如果你已经训练了一个 SSD 模型，并且想要打印出其 Precision，可以按照以下步骤进行： 1. 首先，确保你已经加载了 PyTorch 库和 SSD 模型。 2. 然后，加载测试数据集，并使用模型对其进行预测。可以使用以下代码： ``` from datasets import COCODataset, VOCDataset from ssd import build_ssd import torch # 加载测试数据集 test_dataset = COCODataset(root='/path/to/coco/', image_set='val2017', transform=None, target_transform=None, dataset_name='COCO') # 加载 SSD 模型 net = build_ssd('test', 300, test_dataset.num_classes) net.load_state_dict(torch.load('/path/to/weights.pth')) # 对测试数据集进行预测 net.eval() for i in range(len(test_dataset)): image, target = test_dataset[i] with torch.no_grad(): detections = net(image.unsqueeze(0)) ``` 3. 接下来，计算 Precision。可以使用以下代码： ``` from eval_utils import calculate_precision # 计算 Precision precision = calculate_precision(detections, target, iou_threshold=0.5) print(precision) ``` 在上面的代码中，`eval_utils` 包含了一个名为 `calculate_precision` 的函数，它可以计算 Precision。`detections` 是 SSD 模型对图像的预测结果，`target` 是图像的真实标注，`iou_threshold` 是计算 Precision 时使用的 IOU 阈值。函数返回的是一个标量，即 Precision 的值。 注意，`calculate_precision` 函数需要使用 Python 3.7 或更高版本。如果你的 Python 版本较低，可以手动计算 Precision。

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值