![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
Soul·Top
soultop
展开
-
python 深度学习图片格式转换的总结
cv2 转 tensorimg = cv2.imread("a.png")# 转成 RGB 形式img0 = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)# 归一化,并添加一个维度(Batch)img0 = torch.from_numpy(img0).div(255.0).unsqueeze(0) # N*W*H*C# 置换维度形式为 N*C*W*Himg = img0.permute(0, 3, 1, 2)cv2 转 numpy(PIL.Image)imp原创 2020-11-27 12:38:06 · 653 阅读 · 0 评论 -
特征提取模型的轻量化之路(一)
一、 参数量和理论计算量1、计算公式:假设卷积核大小为 Kh×Kw{K_h×K_w}Kh×Kw,输入通道数为 Cin{C_{in}}Cin,输出通道数为Cout{C_{out}}Cout,输出特征图的宽和高分别为W{W}W和H{H}H,这里忽略偏置项CONV 标准卷积层:params:Kh×Kw×Cin×Cout{K_h×K_w×C_{in}×C_{out}}Kh×Kw...原创 2019-12-12 21:31:47 · 2191 阅读 · 0 评论 -
Ubuntu16.04,Nvidia驱动+CUDA10.0+cuDNN7.4+CMake+OpenCV3.4.5+Darknet+tensorflow-gpu安装配置流程
1、系统及硬件我的系统是ubuntu16.04,显卡是双2080Ti。因为楼主也是初次基础ubuntu系统,所以在安装这些东西的时候多次把系统搞崩溃,导致系统重复登陆,或是无图形服务。这篇文章我会尽量写详细一些。如果有什么问题可以在讨论。2、Nvidia驱动卸载及安装2.1、下载驱动这里一定要注意!!!!这里一定要注意!!!!这里一定要注意!!!!进入tensorflow官网的Inst...原创 2019-04-06 20:36:20 · 719 阅读 · 0 评论 -
pytorch 1.0实现faster R-CNN,检验自己的数据集
pytorch 1.0实现faster R-CNN简介Step1、安装Step2、下载特征提取层的预训练(pre-trained)权重Step3、训练前的其他准备1)、准备数据集2)、修改相关代码Step4、开始训练Step4、批量Test训练好的model1、修改demo.py文件[1]2、输入指令3、可能遇到的问题及解决方案:简介 这个工程实战是***在Ubuntu***系统上基于jwy...原创 2019-07-22 17:04:28 · 10535 阅读 · 72 评论