![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
python
文章平均质量分 53
碧影江白
读研中
展开
-
普通用户安装局部cuda,cudnn(非默认路径)
之前在服务器的全局环境中的cuda被人瞎动破坏了(很气),现在做下在局部环境里的重装。首先下载好两个文件,下载位置是用户文件的首页地址。cuda和cudnn不多说,版本匹配做好。用xshell登录进入服务器,直接在默认环境下,激活环境:activate tf(环境的名字)。安装cuda:sh cuda_9.2.148_396.37_linux.run这里和全局用户添加不一样的在于没有sudo,因为普通用户没有sudo权限。安装时候的选项:这里的位置需要设置为自己.原创 2021-10-23 17:29:25 · 594 阅读 · 0 评论 -
数据集制作、数据读取等在机器学习时可以用到的
在训练神经网络时,经常需要使用到某些数据制作、读取的代码,这里做个总结。方便以后使用。1、数据保存我制作的数据集,input为已经处理好的txt文件,label为xml文件,在保存时将它们保存为同名文件:以上为保存的数据名称和地址。2、制作数据集:先在相应的位置创建两个txt文件:train.txt、test.txt。然后运行以下代码。以下代码为将数据按照4:1的比例切割为训练集和测试集。import osimport randomxmlfilepath = "C..原创 2021-09-29 20:23:59 · 679 阅读 · 0 评论 -
python手动实现roi_pooling
roi_pooling首先在fast rcnn中被提出。其可以解决在rcnn中直接把图片resize操作所造成的信息丢失的问题。roi_pooling的规则为把任意大小的的特征矩阵转化为指定大小。具体转化规则为:把原矩阵按照一定的比例划分为各个部分,取该部分的最大值,复制到缓冲区,作为目标矩阵的值。很常见且很直观的一组图:很多资料都在使用faster-rcnn时把roi_pooling替换成了功能相似的roi_align,但当想要使用roi_pooling时发现找不到可以供我使用的接口。原创 2021-03-14 20:42:38 · 1017 阅读 · 1 评论