自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+

村民的菜篮子

拾到篮子里都是菜

  • 博客(50)
  • 资源 (10)
  • 问答 (1)
  • 收藏
  • 关注

原创 pytorch模型保存和加载

pytorch,模型保存,加载

2022-11-10 18:53:24 2970 1

原创 知识蒸馏方法汇总

知识蒸馏,互学习

2022-11-10 14:28:26 3045

原创 小目标检测方法总结

小目标检测,切图拼图,SNIPER

2022-11-09 18:56:22 3602 2

原创 YOLOX运行demo及细节理解

1、Focus参考:4、Focus模块-in YOLO - 知乎 (zhihu.com)代码实现:class Focus(nn.Module): """Focus width and height information into channel space.""" def __init__(self, in_channels, out_channels, ksize=1, stride=1, act="silu"): super().__init__()

2021-08-10 09:30:10 7548 7

原创 caffe编译遇到protobuf相关问题及解决方法

3、libprotobuf相关的错误,提示版本不对应实际上时protobuf的版本对应有问题,我编译使用vs2015,但是项目下的protobuf版本是基于vc120编译的,只要找到vc140版本的放进去就可以了同理,如果有boost相关的,找不到libboost_date_time-vc140-mt-1_65_1.lib之类的错误:也是因为使用编译的vs版本不一样导致的,换成对应的vc140版本库既可以了。...

2021-08-05 10:52:06 364

原创 用CenterNet做3D检测,效果demo

一、准备1、环境:win10 x64, pytorch 1.7+python3.7虚拟环境,cuda10.1 update2,GTX10602、centernet官方代码:https://github.com/xingyizhou/CenterNet3、DCNv2最新版本(支持pytorch1.5-1.7):https://github.com/lbin/DCNv2/tree/pytorch_1.74、下载官方预训练模型:ddd_3dop(在KITTI上训练)所有modelzoo里的模.

2021-07-09 11:09:08 1654 8

原创 win10编译DCNv2

环境:win10x64 , vs2019 , pytorch1.7(虚拟环境),python3.7 ,cuda10.1 update2背景:运行centernet demo中的DLA和DCN预训练模型效果,但是DCN需要先编译才能使用编译过程:1、下载DCNv2的github源码:https://github.com/lbin/DCNv2/tree/pytorch_1.7放在centernet的库目录下:这有好几个文件夹:DCNv2是centerNet自带的,算是比较老的..

2021-07-09 08:48:29 3328 3

原创 BVLC版window-caffe编译cpu版本

git地址:https://github.com/BVLC/caffe环境:win10,编译cpu版下载源码,解压:按照github上的readme,先运行git checkout windows,会在scripts目录下生成build_win.cmd:打开build_win.cmd,做如下修改,1、因为只编译cpu版,所以将cuda相关的设为02、WITH_NINJA =03、这里编译python3.5版本,所以python_version需要改成3.接

2021-07-01 18:00:12 322

原创 PytorchToCaffe工具demo测试

pytorch模型转caffe模型的项目地址:xxradon/PytorchToCaffe: Pytorch model to caffe model, supported pytorch 0.3, 0.3.1, 0.4, 0.4.1 ,1.0 , 1.0.1 , 1.2 ,1.3 .notice that only pytorch 1.1 have some bugs (github.com)

2021-06-18 18:04:09 660 1

原创 pytorch冻结网络参数,requires_grad与optimizer顺序的关系

问题说明:pytorch迁移学习时,需要对某些层冻结参数,不参与方向传播,具体实现是将要冻结的参数的requires_grad属性置为false,如下:

2021-06-16 11:39:39 4817 5

原创 tensorboard每次打开都显示旧的日志的问题

问题描述:在运行一个项目生成的tensorboard日志时,可以正常生成浏览器的显示路径:可以每次打开显示的都是其他项目

2021-06-15 14:52:20 2790

原创 TortoiseSVN的状态图标失效的解决办法

借鉴大佬的方法:https://www.cnblogs.com/shizilunya/p/11065521.html还有另外一篇教程里的做法,其实也就是按上面那个大佬的做法做的,我照着做了,但是没成功,记录下踩坑过程。按照大佬的步骤:我第一次没看仔细,直接把中间的数字全部提前了,但是没用,后来发现前面的空格数不够,下面的一连串的Tortoise和前面的其他选项的空格没有对齐,导致修改过后还是没有,最后直接将其他选项的前面的空格直接复制到1Tortoise、2Tortoise、3Torto

2021-03-14 22:48:04 236

原创 多目标跟踪算法JDE在 UA-DETRAC数据集上训练

环境:win10,cuda 10.1 , GTX1060一、数据处理1、数据集获取:链接:https://pan.baidu.com/s/1K3rI9PvzHc1KqOJITNMdVg提取码:lox42、数据集格式数据格式也不一定完全按照上面这种,但是必须得保证图片和标签的名字相同。以MOT17(JDE的Modelzoo中下载得到)为例:文件夹结构:labels_with_ids文件夹里面是用转换工具将gt.txt生成对应的JDE训练所需的标注文件,对应每一个视频序列.

2020-05-19 16:15:13 8435 64

原创 CenterNet caffe版编译和训练自己数据集记录

github上面发现有人实现了caffe版的centernet大佬的github地址:https://github.com/yuqj1990/caffe_train这篇博客记录下我在windows下编译支持centernet的caffe环境的编译过程以及尝试训练和测试。一、编译caffe环境首先看看作者实现的训练centernet所需的loss层:CenterObjectLoss作者提供了:CenternetLossLayer.cpp和CenternetLossLayer.hpp

2020-05-15 15:03:10 1010 2

原创 CVPR 2020 多目标跟踪算法 FairMOT代码解读

总览1、训练(1)数据(2)损失函数(3)跟踪2、测试(1)跟踪(2)跟踪后处理1、训练(1)数据数据集类JointDataset在src\lib\datasets\dataset\jde.py文件中:首先__getitem__方法中读取图片和对应的标签,做简单的数据增强,统计所有数据集的ID最大值,也就是最多有多少个需要跟踪的对象。由于FairMOT的...

2020-05-15 08:52:59 7830 13

原创 YOLOv3 anchor理解

这篇博客记录自己对yolo v3中的anchor怎么用来做目标检测以及特征提取的理解,其他单阶段目标检测中的anchor原理类似。yolo v3中有三个yolo层,

2020-04-25 17:46:51 837 2

原创 CVPR 2020 多目标跟踪算法JDE 训练

数据集:MOT17,取其中的MOT17-13-DPM,MOT17-13-FRCNN,MOT17-13-SDP三个文件夹检测目标:人,车,自行车,摩托车原始数据标注:数据处理:(1)按照原始数据集标注,gt.txt中的倒数第三列是目标轨迹是否进入考虑范围的标志,这里将我们需要的四个类别(人,车,自行车,摩托车)的这个标志设置为1,可以用notepad++打开,查找,替...

2020-04-22 16:48:52 6949 33

原创 CVPR 2020 多目标跟踪算法 FairMOT代码demo运行及训练

1、本机环境win10 64位,cuda 10.1,cudnn 7.5,pytorch 1.4.02、运行demo下载github上的代码:https://github.com/ifzhang/FairMOT以及 预训练模型,这里我下载了HRNetV2-W18网络的预训练代码,因为DLA网络用到了DCN v2,这个需要提前编译,但是我这里编译失败了,作者说编译失败可以用HRNetV...

2020-04-19 11:26:54 19522 134

原创 caffe编译DenseBlock层出错集合及解决办法

无法解析的外部符号 cudaConfigureCall

2020-03-27 11:30:37 766

原创 caffe日志中的Memory required for data

最近在优化轻量化卷积神经网络,需要对模型的内存占用、FLOPs、MACCs等进行分析。在使用caffe框架进行训练时,打印出的日志中有Memory required for data的值,以前没仔细了解过这个值是什么意思,现在得去看看代码,首先得找到这个输出语句是哪里输出的,经过查找发现,这个语句在net.cpp的init()函数中:这个输出的值是:变量memory_used_ * 类...

2020-03-20 15:14:43 643 1

原创 目标检测数据层参数解析(caffe)

数据层名:AnnotatedData1、transform_param作用:数据变换transform_param { # 镜像,左右翻转 mirror: true # 三个通道均值 mean_value: 104.0 mean_value: 117.0 mean_value: 123.0 #将输入图像resi...

2019-12-26 09:35:54 697 1

原创 windows编译pycaffe的采坑点

按照正常步骤,修改commonSettings.props然后再pycaffe项目上点右键,重新生成,会在bin目录下生成pycaffe文件夹和_caffe开头的四个文件将这4个文件复制到pycaffe目录下的caffe文件夹下:将这个caffe文件夹复制到python的site-packages文件夹下:然后在打开python终端,输入import caffe...

2019-12-24 10:37:20 346

原创 目标检测中的priorbox(anchor,default box)可视化(c++)

网络结构,是Yushiqi老师开源的人脸检测网络四个用于检测的特征层信息如下:层 尺寸 min_size(1:1) priorboxes conv3_3 40x40 10,16,24 40x40x3=4800 conv4_3 20x20 32,48 20x20x2=800 conv5_3 10x10 64,96...

2019-12-20 16:01:45 1028 6

原创 vs2013中可视化caffe某层的feature map

记录下自己的实验#include <caffe/caffe.hpp>#ifdef USE_OPENCV#include <opencv2/core/core.hpp>#include <opencv2/highgui/highgui.hpp>#include <opencv2/imgproc/imgproc.hpp>#endif /...

2019-12-20 15:35:56 166

原创 python 读取voc格式的标注xml文件

#coding : utf-8import osimport cv2import abcimport xml.dom.minidom as xmlimport mathimport matplotlib.pyplot as plt'''读取测试集中的xml文件,统计下测试图片中车牌的大小信息,可以用来参考设置priorbox的大小'''class XmlReader(o...

2019-12-20 15:29:49 1157

原创 普通卷积,Depthwise卷积,分组卷积的对比

1、普通卷积如上图所示输入:H*W*c1,卷积核:c1*k*k*c2(卷积核维度为c1*k*k,共有c2个)输出:H'*W'*c2参数量:c1*k*k*c2 (不考虑bias)运算量:c1*k*k*c2*H'*W'2、分组卷积(假设group=2)输入:H*W*c1卷积核:c1*k*k*c2(卷积核维度为c1*k*k,共有c2/2个)输出:H'*W...

2019-11-17 19:02:13 1116

原创 caffe训练日志可视化代码(acc,train_loss,test_loss,lr 画在同一张图中)

caffe自带可视化训练log的脚本程序但是这个程序会将:train_loss,test_loss,lr,acc每个指标单独画一张图(根据参数可选则画什么),如下:我觉得这样生成的图片太多了,而且train_loss和test_loss(实际上是训练过程中的val_loss)通常是要放在一起对比来看的,所以放在同一张图中会更清晰的看出是否过拟合或者欠拟合。因此我自己写了个程序提取...

2019-11-06 09:48:20 2246

原创 caffe训练和测试注意点

训练和测试注意的点:1、在train.prototxt文件中权重和偏置项需要初始化。2、在使用caffe调用自己训练好的模型时,一定要将test.prototxt文件中的权重偏置等参数初始化设置全部删除,要不然会在创建层时卡住退出,这里要删除的内容都是需要反向传播的参数,如卷积和全连接层中的权重和参数初始化以及学习率。3、在使用test.prototxt时,要注意是否需要预处理,如减...

2019-10-25 09:34:06 492 1

原创 finished with exit code -1073740791 (0xC0000409)总结(caffe框架,pycharm)

一、可能原因:(1)caffe协议文件中的某些层不存在(2)deploy文件中的输入层没有添加(3)deploy协议文件中某些层的训练参数没有删除(4)疏忽所致,deploy文件的语法错误,或者某些标点或编码错误。(5)显存不足二、解决办法(1)修改caffe的源文件,添加相关的层,然后重新编译。(2,3)如果是在测试中报错,那就检查deploy.prototxt文...

2019-09-26 14:38:26 602

原创 caffe中添加centerloss层,并在mnist上测试

一、添加centerloss层可参考github上的caffe工程:https://github.com/ydwen/caffe-face添加hpp,cpp,cu文件,以及修改caffe.proto文件重新编译。二、在mnist数据集上使用centerloss训练模型,测试效果下面训练和测试的文件都是在上面的github中。1、训练协议文件:mnist_train_test.p...

2019-09-23 17:43:12 456

原创 vs2013调用RFCN来输出分类概率

环境:win10 x64,vs2013, caffe(带有psroipooling层)测试代码已经传到百度网盘:链接:https://pan.baidu.com/s/1Yy2gnd_g3Jj0uV1NV5eqPg提取码:vp3t复制这段内容后打开百度网盘手机App,操作更方便哦备注:这份代码主要是想在vs2013中调用matlab中训练好的RFCN模型来计算一张图像中各个ROI的概...

2019-07-16 18:29:05 190

原创 运行py-RFCN的demo

代码下载链接:https://github.com/YuwenXiong/py-R-FCN下载预训练模型:https://pan.baidu.com/s/1o77gFXo环境安装,caffe,编译pycaffe接口,python2.7将预训练模型放在dada文件夹下的rfcn_models文件夹,没有就新建:然后开始运行tools文件夹下的demo_rfcn.py文件下...

2019-07-10 09:28:20 637 1

原创 多目标跟踪系列算法:DMAN

下载源码:https://github.com/jizhu1023/DMAN_MOT下载训练好的模型,新建文件夹model,将模型文件放在model文件夹下:作者使用的是python2.7,我用3.5,所以会报print错误,全部加上括号即可。然后在pycharm中运行calculate_similarity.py在matlab中将当前目录切换到DMAN_MOT-mas...

2019-07-09 16:28:39 3041 7

原创 测试训练好的siamese网络

在这篇博客中https://blog.csdn.net/sinat_33486980/article/details/91440461新训练了单分支的siamese网络。参考了论文:Online Multi-Object Tracking with Historical Appearance Matching and Scene Adaptive Detection Filtering结...

2019-06-21 17:46:22 1573 19

原创 caffe保存训练日志并绘制loss,acc曲线

环境windows10 x64+caffe1、导出训练日志到log文件start.bat是我训练网络使用的 批处理文件start.bat的内容是:因为我把caffe加到了系统path中,所以这里直接写caffe train。后面的--solver 是训练用到的协议文件,定义一些参数,如指定网络定义文件,指定学习率,模型保存位置,学习衰减参数,以及优化算法等。最后的这个 &g...

2019-06-20 15:46:00 1067

原创 安装pycocotools

环境:windows10x64 ,python 3.5.2 ,vs2013,单独安装了vc++2015 build tools。在python命令行下输入pip install pycocotools可以直接安装上pycocotools 2.0,但是这是假象虽然可以导入pycocotools,但是导入pycocotools._mask时就会出错,提示不存在这个module,所以需...

2019-06-19 18:47:09 21703 17

原创 tensorflow 1.13.1 安装采坑

环境:win10 x64位,cuda10.1,cudnn 7.5,vs2013,vs2015 distributed ,GTX1060按照网上的教程安装,如下面博客https://blog.csdn.net/huanyingzhizai/article/details/89298964我最后的安装位置:anaconda2下面的虚拟环境py3下面新建虚拟环境:tensorflow-g...

2019-06-19 17:30:36 10967 1

原创 windows下几种深度学习平台检查是否可以成功调用GPU加速

前提是正确安装cuda和cudnn,以及对应的平台。1、caffecaffe有个命令行参数:device_query 可以查看指定GPU的信息如:在命令行输入 caffedevice_query -gpu 02、pytorch直接在python环境下输入以下命令,显示true就表示可以调用gpuimport torchprint (torch.cuda.is_...

2019-06-19 10:57:22 6193 1

原创 R-FCN理解

参考博客:https://www.cnblogs.com/lillylin/p/6277094.html整个R-FCN的流程:一个base的conv网络如ResNet101, 一个RPN(Faster RCNN来的),一个position sensitive的prediction层,最后的ROI pooling+投票的决策层如图所示:下面是我自己画的图,从经过特征提取网络得...

2019-06-15 21:24:43 337

原创 caffe训练siamese网络(行人数据集)

参考博客:https://blog.csdn.net/gybheroin/article/details/54133556#commentsedit环境:windows10 x64 ,caffe, cuda 10.1,cudnn 7.5数据集:行人数据集文件夹目录结构:----------------------------------数据准备---------------...

2019-06-11 15:50:43 1545

Darklabel图片视频标注工具.zip---(内含详细操作说明和处理脚本)

这个工具可以对连续的图片集或者一个视频进行半自动化标注,可以用来标注目标检测和多目标跟踪需要的数据,所谓半自动化,是针对多目标跟踪数据集的标注来说的,就是不用每一帧都重新标目标,且能通过内置的多目标跟踪算法自动预测出下一帧的目标位置并与上一帧的目标编号一一对应,只需手动对自动预测的框做微调即可,非常快捷,亲测一小时标注500-1000张完全没问题。

2020-06-18

YuDetectNet.zip

Yushiqi老师的轻量化人脸检测网络(带预训练模型2.6M),v1版将输入图像resize到320x320,v2版输入图像resize到320x240。测试检测效果非常好

2019-12-20

L-softmax.zip

caffe中使用L-softmax代替softmax损失,在mnist数据集上测试,并画出最后的测试效果图。

2019-12-20

caffe网络结构可视化工具-netron

查看神经网络结构的利器,清晰漂亮,比Netscope更好看,还可以放大缩小,不用粘贴代码,安装完之后就会默认使用netron打开prototxt文件,用了很久,良心推荐

2019-11-20

mixnet.zip

官方说超越MobileNet v3的MixNet,使用混合尺度的Depthwise卷积,效果提升明显。本人对照pytorch版本实现的caffe版本,包括s和m两个版本。在200类ImageNet数据集上测试top1准确率为0.83。

2019-11-13

YOLO_v3_pytorch.zip

目前测试过的yolo v3 pytorch版本的网络上的代码,修改了图片保存代码,可以完美运行,附带我的运行结果以及打印出的网络结构,一目了然

2019-11-13

EfficientNet-Caffe.zip

神经网络架构搜索出的网络EfficientNet,根据官方开源pytorch代码,转为caffe的网络协议文件,可以顺利训练,这里提供了EfficientNet B0的3种版本,第一种是完整 版,也就是使用了swish激活以及SE模块的,第二种是将swish替换为relu,且包含SE模块,第三种是将swish替换成relu,且去掉SE模块的最精简版本。经过在我自己的250类车辆分类数据集上,精简版可以达到0.9992的准确率,relu+SE版的准确率为0.9968,完整版本的还没训练结束,不过相差不多,效果很不错。

2019-10-22

MoGA-caffe.zip

小米的神经网络架构搜索出来的MoGA网络,包括A、B、C三个版本,通过对比官方开源的pytorch版本,实现的caffe网络协议文件,可以正常训练。

2019-10-22

mobileNetv3.zip

包含MobileNet v3 large和small两个版本,其中除了原始版本的使用H-swish和SE模块的prototxt外,还有用relu和SE模块实现的版本,以及使用relu并去掉SE模块的最精简的版本,三个版本测试效果差别不大,最精简的版本,模型最小(5.89M),我自己得250类车辆分类数据集上可以达到0.996的测试准确率。

2019-10-22

SigAI系列的机器学习和深度学习总结好文

SigAI系列出的总结的有关机器学习和深度学习一系列文章,包括需要的一些数学基本知识,以及经典算法的推导和扩展,内容对找人工智能相关工作的人或者入门机器学习深度学习的人来说,都很有用,推荐学习!

2019-03-17

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除