深度学习
tensorflow
碧影江白
读研中
展开
-
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 · 1046 阅读 · 1 评论 -
tensorflow读取一个模型后多次使用
训练好一个模型后,将其投入使用,会有在项目初始化后多次加载测试数据的需求,可以采用保存graph的思想实现(在一个项目中需要加载多个模型同样可用)另:这条博客接我的上一条https://blog.csdn.net/qq_34470213/article/details/104076898,是在上一个代码的基础上改写的。1、写一个初始化函数,用来初始化模型(项目中仅需初始化时调用一次)class Test(): def restore(self): self.mode原创 2020-07-06 20:22:45 · 2273 阅读 · 2 评论 -
python + tensorflow 神经网络多分类
新电脑的所有项目资料都没了,现在想搞一个简单的CNN多分类问题都还得重新写好麻烦,简单记录一下实现步骤,用于今后备用。1、以服装分类为例,在百度图片上搜了各类衣服的图片,裁成方块,分成5类,每一类都放到一个文件夹,为了方便取名0-42、裁剪成固定尺寸并且镜面旋转后阈值化操作(这里采用的是自适应阈值,也可不使用阈值化,灰度图亦可)重新保存。(原图路径D:/image,保存路径D...原创 2020-07-04 18:44:17 · 1945 阅读 · 0 评论