- 博客(17)
- 资源 (1)
- 收藏
- 关注
原创 Pytorch目标检测学习资源
Pytorch学习陈云《深度学习框架PyTorch:入门与实践》:https://zhuanlan.zhihu.com/p/31712507官方教程:http://pytorch123.com/动手学深度学习:https://zhuanlan.zhihu.com/p/85353963其他开源资料:https://github.com/zergtant/pytorch-handb...
2019-09-30 10:00:53 5261
原创 代码开源!!行人检测与行人重识别结合 person search
0 前言最近在做自己课题相关的小项目,行人检测与行人重识别进行结合进行场景图片进行特定行人的检索由于也比较着急,先利用现有的模型搭建了简单的demo,简单效果展示如下,速度大概在20FPS:先是给定需要检索的行人:在监控场景图片进行检索该行人:还有视频的demo展示:利用下面一张其他摄像头拍摄的特定行人图片在视频中进行搜人:https://www.bilibil...
2019-09-29 11:09:23 11152 10
原创 SKU110K商品检测数据集处理
1. 找出损坏的图片,形成junk.txttest_1346.jpgtest_184.jpgtest_2643.jpgtest_274.jpgtest_2878.jpgtrain_2376.jpgtrain_2986.jpgtrain_305.jpgtrain_3240.jpgtrain_38.jpgtrain_5007.jpgtrain_5137.jpgtrain_5822.jpgtrain_737.jpgtrain_775.jpgtrain_789.jpgtrain
2020-12-23 10:28:19 2581 1
原创 DeepFashion2 数据集
论文标题:DeepFashion2: A Versatile Benchmark for Detection, Pose Estimation, Segmentation and Re-Identification of Clothing Images数据集链接:https://github.com/switchablenorms/DeepFashion2简单介绍:https://www.jiqizhixin.com/articles/2019-02-04-2数据集下载解压解压需要密码,不过
2020-05-18 14:08:11 4189 4
原创 7x7换成3个3x3卷积
速度上首先写了下面的一段代码分别在CPU和GPU上进行测试时间import torchimport torch.nn as nnimport torchvision.models as modelsclass BasicConv(nn.Module): def __init__(self, in_planes, out_planes, kernel_size, stri...
2020-03-30 15:30:30 7125 12
原创 行人重识别数据集转换--统一为market1501数据集进行多数据集联合训练
0前言常用的reID数据集如图所示下载好数据集,我的数据集一开始是这样的第一步创建出来market1501的数据集文件夹格式market1501数据集的具体介绍可以看看这个http://blog.fangchengjin.cn/reid-market-1501.htmlimport osdef make_market_dir(dst_dir='./'): ...
2019-11-09 17:04:08 11736 14
原创 WiderPerson行人检测数据集
1.简介WiderPerson数据集是比较拥挤场景的行人检测基准数据集,其图像是从多种场景中选择的,不再局限于交通场景。 选择13382张图像,并用各种遮挡标记约40万个注释。 我们随机选择8000/1000/4382图像作为训练,验证和测试集。 与CityPersons和WIDER FACE数据集相似,不发布测试图像的标注文件。可以在官网上进行下载:http://www.cbsr.i...
2019-10-26 15:37:42 17311 11
原创 对YOLOv3进行瘦身-行人检测
0 前言前面用RFSong实现了GPU 200FPS,并且测试结果发现在COCO上也只比YOLOv3低了六七个点左右,基本可以满足自己的使用需求了。因此下面想试试YOLO按照RFSong进行改进,看看能不能在精度只掉一点的情况下,实现更快的速度。1.确定基线尝试对YOLO在VOCCOCO的行人图片数据集进行从头训练,也就是不加载任何权重直接进行目标检测任务的训练(这里没有开启多尺...
2019-10-24 08:57:19 4057 1
原创 3. 检查下转换YOLO数据集是否成功
import osimport cv2img_formats = ['.bmp', '.jpg', '.jpeg', '.png', '.tif']if __name__ == '__main__': classes = ['person'] path = '/home/wangsong/data/val2017.txt' with open(path, 'r'...
2019-10-23 11:25:41 352
原创 2. COCO转YOLO--只要行人
# coding=utf-8from pycocotools.coco import COCOimport shutilimport os# 将ROI的坐标转换为yolo需要的坐标# size是图片的w和h# box里保存的是ROI的坐标(x,y的最大值和最小值)# 返回值为ROI中心点相对于图片大小的比例坐标,和ROI的w、h相对于图片大小的比例def convert(si...
2019-10-23 11:23:56 2599 4
原创 1. VOC0712转YOLO-只要行人
简介大部分都是参考YOLO从零开始:基于YOLOv3的行人检测入门指南,不过他这个还得分几步,各种配置也比较麻烦,因此进行了优化,直接一步到位。具体操作原来的VOC2007和VOC2012的路径是这样的:建立一个data文件夹,这样直接按照相对路径去写图片和标注文件位置,方便自己使用 运行下面代码import xml.etree.ElementTree as ETimp...
2019-10-22 19:19:16 671 1
原创 YOLOv3学习率
YOLOv3官方版本的学习率配置信息在模型配置文件 *.cfg file 中:learning_rate: 标准学习率 burn_in: 学习率从 0 上升到 learning_rate 的 batch 数目 max_batches: 需要进行训练的 batch 数目 policy: 学习率调度的策略 steps: 在何处进行学习率衰减 scales: 学习率进行衰减的倍数这个...
2019-10-21 08:58:29 7075 1
原创 Rectangular inference 矩形推理
前言学习YOLOv3代码https://github.com/ultralytics/yolov3Rectangular inference 矩形推理是detect.py中的默认实现。这可以显著的减少推理时间。 例如,在zidane.jpg上,CPU推理时间(在2018年MacBook Pro上)从1.01s减少到0.63s,减少了37%。这是因为图像区域送入网络时减少了38%(416x...
2019-10-19 17:18:16 6203 11
原创 将行人检索的PRW数据集转为VOC--行人检测
import osimport numpy as npimport scipy.io as sioimport shutilfrom lxml.etree import Element, SubElement, tostringfrom xml.dom.minidom import parseStringdef make_voc_dir(): # labels 目录若不存...
2019-10-13 19:06:17 1341 2
原创 将SYSU数据集转换为VOC--只关注行人检测
1.构建VOC文件夹import osdef make_voc_dir(): # labels 目录若不存在,创建labels目录。若存在,则清空目录 if not os.path.exists('../VOC2007/Annotations'): os.makedirs('../VOC2007/Annotations') if not os....
2019-10-13 16:25:37 857 6
原创 COCO转VOC---提取行人图片和标注
在只用VOC数据集训练得到的行人检测效果还是泛化效果一般,因此接下来尝试将多个数据集转成YOLO格式和VOC格式。from pycocotools.coco import COCOimport shutilimport osfrom lxml.etree import Element, SubElement, tostringfrom xml.dom.minidom import p...
2019-10-11 22:54:51 1968 1
原创 person search代码使用小教程
0前言前面开源了行人检索的代码。由于是多数据集联合训练,因此泛化效果还是不错的,一些简单场景都可以直接用。具体可以看看下面,可爱的实验室师兄师姐们拍摄的演示视频进行测试的结果。下面具体介绍一下,这个演示视频是具体怎么进行测试的:1.截取得到待检索的行人图片对于侦探寻找犯人来说,往往是有一个中心现场的监控得到了犯罪嫌疑人的图片。因此,我们先利用query_get.py截取中心...
2019-09-29 15:41:59 13828 45
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人