深度学习
Crazy-Monica
这个作者很懒,什么都没留下…
展开
-
安装mmcv-full适配torch版本
比如我的cuda版本是10.1,torch版本是1.8.0,mmcv-full安装命令如下。原创 2023-02-08 17:31:02 · 5615 阅读 · 0 评论 -
ubuntu14.04+caffe+cuda8.0+openpose工作环境搭建
近期要用到openpose,原创 2017-07-20 20:38:20 · 8029 阅读 · 11 评论 -
生成非图像类型的LMDB数据
最近在训练网络中会用到非图像类型的数据,我这里是将这种数据转换成LMDB类型作为一个数据层,加载进网络。主要用到caffe的Python接口。 1、在网络的中间层中,其接受一个1x6维的bottom数据作为输入; 2、每个训练样本对应的1x6维的数据存储到data.txt,同时记录其类别标签; 3、写入LMDB 。#-*- coding: UTF-8 -*-import numpy a...原创 2018-03-15 18:14:05 · 536 阅读 · 0 评论 -
调用caffemodel中的权重赋值给自己的网络
调用预训练好的caffemodel,将其权重赋值给自己网络中的对应层。基本思路比较简单,三步走: 1、加载预定义的caffemodel; 2、加载要训练的网络; 3、将预训练model的参数赋值给要训练的网络。import caffeimport numpy as npimport math# 加载预训练网络net_ = caffe.Net("deploy.prototxt...原创 2018-03-16 14:00:44 · 2406 阅读 · 0 评论 -
Python--3类图像每个类别中随机挑选n张图片(小白程序)
深度学习中经常会遇到要从好几类图片中随机挑选部分数据训练或者测试的问题:举个栗子:现有训练的标签文件train.txt,里面包含3类图片,现在要在这3类中每个类别随机选取10张train.txt的内容比如:…/media/data/train/1.jpg 1/media/data/train/88.jpg 2/media/data/train/677.png 0…code:...原创 2019-06-26 16:50:36 · 2508 阅读 · 0 评论 -
Python统计标签个数
深度学习中的图像处理,包含多个标签的数据集,往往存在类别不平衡的问题,这样会导致训练模型准确率低的问题,因此有必要统计,各个类别图片标签个数,以及所占的比例。举例:现有标签文件train.txt,内容如下:/data/train/1.jpg 4/data/train/88.jpg 2/data/train/455.jpg 1/data/train/764.jpg 3/data/trai...原创 2019-07-08 09:30:44 · 6837 阅读 · 0 评论 -
tensorboardX 报错TypeError: __init__() got an unexpected keyword argument 'seserialized_options'
from tensorboardX import SummaryWriter报错如下:TypeError: __init__() got an unexpected keyword argument 'serialized_options'解决方法:set PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=cpppython -mpip install pro...原创 2019-07-18 14:56:15 · 1372 阅读 · 0 评论