python
xiaoju233
Talk is cheap, show me the code.
展开
-
pytorch手把手使用FasterRCNN进行目标检测
手把手使用FasterRCNN进行目标检测使用pytorch自带的预训练FasterRCNN目标检测模型进行目标检测,并把检测到的结果打印到原图片上。由于pytorch自带的FasterRCNN是使用COCO数据集进行预训练得到的,所以我们需要使用COCO的标签来匹配模型输出的结果。详细的注释都在代码里面了。import torchimport torchvisionfrom torchvision.models.detection.roi_heads import fastrcnn_loss原创 2021-10-06 20:49:13 · 5381 阅读 · 6 评论 -
目标检测算法SSD-detector(pytorch实现)
目标检测算法SSD-detector(pytorch实现)一批图像输入SSD深度神经网络之后,输出的是预测的bboxes的偏移量和每一个类别的置信度,需要对这些输出进行处理才能得到最终的预测结果。对每一张图片的预测值分别进行如下处理:对每一个bbox取其分数最高(经过softmax)的一类作为其类别(注意要先把背景的分数剔除)。取分数排名前k个bbox进行解码。对这些bbox进行多类别nms处理(非极大性抑制)多类别非极大性抑制:考虑到非极大性抑制后,bbox最高得分的类别可能会被剔除,原创 2020-07-14 10:40:21 · 1373 阅读 · 3 评论 -
目标检测评估指标mAP的计算-python
目标检测评估指标mAP的计算-python目标检测性能评估指标mAP介绍为解决不同场景下对目标检测的位置偏差的需求不同,通常给定一个IOU阈值,超过此阈值则视为检测成功。以及考虑到类别平衡的问题,通常分别求每一个类别的性能,再进行类别间求平均。那么给定一个IOU阈值以及一个特定的类别,如何求这个类别的AP值呢?首先对所有的检测结果排序,得分越高越靠前,然后依次判断检测是否成功。先求出检测结果dets和真实目标gts的iou值,并找出每一个det有最大iou值的gts,若这个最大的iou超过了一定的阈原创 2020-07-13 16:16:42 · 2463 阅读 · 3 评论 -
pytorch使用指定的GPU
###pytorch使用指定的GPU在实践中,往往会有一些显卡被其它任务占用,这时我们需要指定使用空闲的GPU进行工作。通过环境变量指定GPUexport CUDA_VISIBLE_DEVICES=0,1,2,3在程序开头设置os.environ[“CUDA_VISIBLE_DEVICES”] = ‘0,1,2,3’在命令行执行的命令前添加CUDA_VISIBLE_DEVICES=0,1,2,3,4使用torch.cuda接口torch.cuda.set_device(0)亲测貌似不行原创 2020-06-13 23:48:19 · 199 阅读 · 0 评论 -
python爬虫爬取微博知乎热搜榜
python爬虫爬取微博知乎热搜榜使用python编写脚本爬取微博和知乎的热搜榜单,原理非常简单,构造http协议get请求(注意:get请求头部中需要包含cookie,否则无法获取到热搜数据,可以先使用浏览器访问,从开发者工具中找到对应cookie,并记录下来,以后一直使用这个cookie即可),再使用request工具包发送get请求,使用lxml工具解析返回的html,从中找出我们想要的数据即可。请求连接微博:https://weibo.com/a/hot/realtime知乎:https:/原创 2020-06-12 13:05:36 · 1737 阅读 · 1 评论 -
python日志模块logging
python使用日志模块logging在python项目中可以使用logging模块来管理记录的日志。简单配置使用basicConfig()方法简单地配置logger即可满足基本的需求。basicConfig()方法参数如下:参数名称参数说明filename日志输出文件的文件名filemode打开文件的模式,有r(+)、w(+)和a(+)format...原创 2020-04-10 23:25:10 · 129 阅读 · 0 评论