随笔分类 -  深度学习

pytorch中构建模型
摘要：pytorch构建层的方法和tensorflow基本一致。 ###1 简单的模型构建 import torch from torch import nn from torch.nn import functional as F class NewModule(nn.Module): def __in 阅读全文
posted @ 2021-11-09 17:58 cyssmile 阅读(117) 评论(0) 推荐(0)
YOLO系列:YOLO v1
摘要：1 YOLO v1 YOLO v1中将图像分为S*S格子，每个格子预测B个boundingbox（对于一个bbox有坐标(x,y), w,h和该bbox的置信度），C个分类得分。 在论文中S=7，C=20,B=2,所以输出的tennsor大小为7*7*30，其中30 = （2*（4+1）+20）。 阅读全文
posted @ 2021-10-19 14:40 cyssmile 阅读(379) 评论(0) 推荐(0)
RetinaFace在TensorRT8.2上部署
摘要：github地址 模型 MoblieNetV1-Retinaface 开发环境 Windows10 CUDA11.1 + CUDNN8.2.1 TensorRT8.2EA GTEST + GLOG OPENCV4 CMake 语言 C++ 代码下载后需要自行配置CUDA和CUDNN，并配置OPENC 阅读全文
posted @ 2021-10-14 02:12 cyssmile 阅读(422) 评论(0) 推荐(0)
日常记录-使用TensorRT错误（二）
摘要：将本机电脑上TensorRT6升级到TensorRT8后，原来的模型报以下错误 ERROR: ModelImporter.cpp:574 In function importModel: [4] Assertion failed: !_importer_ctx.network()->hasImpli 阅读全文
posted @ 2021-10-12 18:39 cyssmile 阅读(1942) 评论(0) 推荐(0)
VOC2007转COCO数据集
摘要：import xml.etree.ElementTree as ET import os import json coco = dict() coco['images'] = [] coco['type'] = 'instances' coco['annotations'] = [] coco['c 阅读全文
posted @ 2021-10-06 15:24 cyssmile 阅读(364) 评论(0) 推荐(0)
cudnn8.0+cuda10.0链接地址
摘要：cudnn真的不好下，这里我下载的链接。 https://盘.摆度.抗/s/1txEtdIfRMNi07NAndONFJQ 密码: jub9 阅读全文
posted @ 2021-01-07 14:42 cyssmile 阅读(1043) 评论(0) 推荐(0)
图像检测算法
摘要：分类：通常图像分类并没有什么用处，只是得出一张图片里面有什么。 定位＋分类：知道图片中有个什么，也把这个物体定位出来了，但是也没啥用，因为日常生活中一张图片中可能有多个物体。 物体检测：做到这一步在实际中就有用处了。 IOU（交并比） 用于衡量定位的准确度，　一般IOU　>= 0.5 可认为定位成功 阅读全文
posted @ 2020-08-27 20:34 cyssmile 阅读(6771) 评论(0) 推荐(0)
基于CNN的图像分类算法
摘要：*　１　对卷积神经网络的研究可追溯到1979和1980年日本学者福岛邦彦发表的论文和“neocognition”神经网络。 *　２　AlexNet使用卷积神经网络解决图像分类问题，在ILSVR2012中获胜并大大提升了state-of-start的准确率（大概16%左右）。（在11年top5的错误率 阅读全文
posted @ 2020-08-27 17:12 cyssmile 阅读(6418) 评论(0) 推荐(0)