Caffe2
文章平均质量分 66
AIHGF
完美一切 一切完美
展开
-
Caffe 与 Caffe2
Caffe 与 Caffe2原创 2017-05-18 11:51:07 · 3106 阅读 · 0 评论 -
Caffe2 - (十七) 基于 FashionMnist 数据集的 GPU 训练与测试
Caffe2 - 基于 FashionMnist 数据集的 GPU 训练与测试 数据准备:下载数据:Fashion-Mnist实现过程:#/usr/bin/env python# --*-- coding: utf-8 --*--import osimport gzipimport numpy as npimport cPickleimport matplotl原创 2018-01-27 10:15:42 · 1865 阅读 · 3 评论 -
Caffe2 - (十八) 图片数据处理函数
Caffe2 - 图片数据处理函数Caffe2 提供了对图片进行加载、裁剪、缩放、去均值、batch 等处理的函数 - helper.py.原创 2018-01-29 15:08:45 · 1017 阅读 · 0 评论 -
Caffe2 - (十九) 基于 Detectron 的 DeepFashion 服装 bbox 检测实现
基于 Detectron-caffe2 的 DeepFashion 服装 bbox 检测实现简单进行实现,主要是数据处理,训练过程不复杂. 这里基于:Detectron 中的 Faster R-CNN 框架,ResNet50 网络;DeepFashion 服装 bbox 标注数据1. 数据处理COCO 数据集格式如 - Dataset - COCO Dataset 数据...原创 2018-03-08 18:14:12 · 3918 阅读 · 10 评论 -
Caffe2 - (二十) Detectron 之 config.py 文件参数
Caffe2 - (二十) Detectron 之 config.py 文件给出了 Detectron 的默认参数,其位于 lib/core/config.py. 类似于 Faster R-CNN 中对应的形式.一般不更改该文件参数,可以根据设置 yaml 文件并利用 merge_cfg_from_file (yaml_file) 来加载自定义参数原创 2018-03-22 13:44:39 · 7084 阅读 · 7 评论 -
Caffe2 - (二十一) Detectron 之模型训练及数据加载流程
Caffe2 - (二十一) Detectron 之模型训练及数据加载流程根据训练过程 train_net.py,大致梳理下 Detectron 的训练过程和训练时数据加载过程.模型训练训练数据集/读取...原创 2018-03-23 21:15:17 · 4082 阅读 · 29 评论 -
Caffe2 - (二十八) Detectron 之 modeling - model_builder
Caffe2 - (二十八)Detectron 之 modeling - model_builderDetectron 支持很多种模型类型,Fast R-CNN,RPN Only,Faster R-CNN,Mask R-CNNRFCN,Retinanet原创 2018-04-08 12:57:41 · 1984 阅读 · 0 评论 -
Caffe2 - (二十二) Detectron 之数据集加载与处理函数
Detectron 是基于标准 COCO json 数据集格式进行的.如果处理新的数据集时,强烈推荐将数据集转化为 COCO json 格式,重用先有数据代码即可.不推荐重写新数据集格式的代码.原创 2018-03-26 11:18:12 · 3439 阅读 · 0 评论 -
Caffe2 - (二十九) Detectron 之 modeling - 基础网络
Caffe2 - (二十九) Detectron 之 modeling - 基础网络VGG16, VGG_CNN_M_1024,ResNet,ResNeXt.1. VGG16.py原创 2018-04-08 17:50:27 · 1532 阅读 · 2 评论 -
Caffe2 - (三十) Detectron 之 modeling - 模型_heads
Caffe2 - (三十) Detectron 之 modeling - 模型_headsfast_rcnn_headds.py,rpn_heads.py,mask_rcnn_heads.py,keypoint_rcnn_heads.py,retinanet_heads.py,rfcn_heads.py原创 2018-04-08 21:37:52 · 1503 阅读 · 2 评论 -
Caffe2 - (三十一) Detectron 之 modeling - FPN 与 optimizer
Caffe2 - (三十一) Detectron 之 modeling - FPN 与 optimizerFPN 模块 - Feature Pyramid Network (FPN) 使用的相关函数,optimizer.py,generate_anchors.py原创 2018-04-09 11:09:39 · 2770 阅读 · 8 评论 -
Caffe2 - (三十二) Detectron 之 roi_data - 模型 minibatch blobs
Caffe2 - (三十二) Detectron 之 roi_data - 模型 minibatch blobs根据对应的 roi_data 模块可以处理 对应模型的 minibatch blobs.fast_rcnn.py,mask_rcnn.py,keypoint_rcnn.py,rpn.py,retinanet.py,原创 2018-04-09 15:08:07 · 2376 阅读 · 3 评论 -
Caffe2 - (二十五) Detectron 之 utils 函数(3)
Caffe2 - (二十五) Detectron 之 utils 函数(3)lr_policy.py,net.py,subprocess.py,timer.py,logging.py,vis.py,coordinator.py 多线程,collections.py原创 2018-04-03 14:43:25 · 1704 阅读 · 1 评论 -
Caffe2 - (二十三) Detectron 之 utils 函数(1)
Caffe2 - (二十三) Detectron 之 utils 函数(1)blob.py,boxes.py ,image.py原创 2018-04-02 14:32:48 · 1445 阅读 · 2 评论 -
Caffe2 - (二十四) Detectron 之 utils 函数(2)
Caffe2 - (二十四) Detectron 之 utils 函数(2)env.py,c2.py,io.py,keypoints.py,segm.py原创 2018-04-02 17:00:08 · 2379 阅读 · 0 评论 -
Caffe2 - (三十三) Detectron 之 roi_data - data loader
Caffe2 - (三十三) Detectron 之 roi_data - data loader1. loader.pyDetectron data loader.py 是通用设计,与 minibatch 的实现细节无关. 多线程的数据加载.minibatch 是字典形式,存储了 blob name keys 和对应的 numpy (float32 或 int32) ndarray values.原创 2018-04-10 11:37:05 · 2911 阅读 · 0 评论 -
Caffe2 - (二十六) Detectron 之定制 Python Operators(ops)
Caffe2 - (二十六) Detectron 之定制 Python Operators(ops)Detectron 基于 Caffe2,添加了三个自定义的网络层(Python Ops).collect_and_distribute_fpn_rpn_proposals.pygenerate_proposal_labels.pygenerate_proposals.py原创 2018-04-04 15:34:13 · 1233 阅读 · 0 评论 -
Caffe2 - (十一)ResNet50 Multi-GPU 训练
Caffe2 - Multi-GPU 训练1. 概要ResNet50 modelImageNet 数据集 - 14 million 张图片, 大概需要 300GB SSD 存储空间,2000 个磁盘分片;两张 GPUs 耗时一周.这里以 ImageNet 中的一部分为例:640 种 cars 和 640 种 boats 图片集作为训练数据集;48 种 cars 和 4...原创 2018-01-10 13:10:46 · 8601 阅读 · 9 评论 -
Caffe2 - (十六) 创建 LMDB 数据库
Caffe2 - 创建 lmdbCaffe2 提供了将数据转换为 lmdb 的 Demo.@package lmdb_create_example# Module caffe2.python.examples.lmdb_create_example原创 2018-01-26 15:41:52 · 1679 阅读 · 0 评论 -
Caffe2 - (十五) 简单的回归示例 Toy Regression
Caffe2 - 简单的回归例子 Toy Regression回归问题:假设二维输入 xxx,一维输出 yyy,权重向量 w=[2.0,1.5]w=[2.0,1.5]w = [2.0, 1.5],偏置 bias b=0.5b=0.5b=0.5,y=wx+by=wx+by = wx + b 注: 这里训练数据是利用 Caffe2 Op 生成的. —— [Toy Regr...原创 2018-01-26 14:33:56 · 1212 阅读 · 0 评论 -
Caffe2 - (十二)自定义 Operators
Caffe2 - Operators 自定义Caffe2 提供了很多 Operators - Operators Catalogue. 如果实际应用中需要自定义新的 Operators,其实现如下.1. 自定义 Basic Operator一般情况下,每个 Operator 包括两个文件:.cc - 注册 operator (registering of the oper原创 2018-01-11 22:30:26 · 2034 阅读 · 3 评论 -
Caffe2 - (一)Source 安装及问题解决
Caffe2 - From Source 安装1. 安装平台 - Ubuntu14.04 - Ubuntu16.04从Source安装依赖项:sudo apt-get updatesudo apt-get install -y --no-install-recommends \ build-essential \ cmake \原创 2017-05-18 11:21:18 · 6733 阅读 · 0 评论 -
Caffe2 - (二)Distributed Training分布式训练
Caffe2 Distributed Training分布式训练Caffe2 最重要的一个特点是,便于分布式训练,无须重构设计即可快速实现网络的伸缩. 进一步学习Caffe2分布式训练的例子,可参考SynchronousSGD, 该材料阐述了Caffe2的data_parallel_model设计规则. Gloo - 用于多机训练的通信库 NCLL - NVIDIA推出的原创 2017-05-18 16:35:11 · 6222 阅读 · 0 评论 -
Caffe2 - (三) Blobs,Workspace,Tensors等概念
Caffe2 涉及的概念1. Blobs,Workspace,TensorsCaffe2 的 Data 是以 blobs 的形式组织的. blob 即是内存中被命名的 data chunk(数据块). blobs 一般包含一个 tensor(可以看做是多维数组),在 Python 中的存在形式是 numpy arrays.Workspace 存储所有的 blobs. 如下例原创 2017-12-29 10:42:34 · 3490 阅读 · 0 评论 -
Caffe2 - (五)Workspace Python API
Caffe2 - Workspace Python APIModule caffe2.python.workspace. Demo:from caffe2.proto.caffe2_pb2 import NetDeffrom caffe2.python import workspaceinit_net = NetDef()init_net.ParseFromString(o原创 2018-01-02 15:52:08 · 2692 阅读 · 0 评论 -
Caffe2 - (六)CPU/GPU 模式切换
Caffe2 - CPU/GPU 部署模式切换在尝试把 Caffe 模型转换到 Caffe2,部署时 CPU/GPU 模式切换方法找了很久才找到一个用着可以,记录下.import syssys.path.insert(0, '/path/to/caffe2/build')from caffe2.python import core, workspace, model_helper原创 2018-01-02 16:06:24 · 5740 阅读 · 2 评论 -
Caffe2 - (七)Caffemodel 转换为 Caffe2 pb 模型
Caffe2 - Caffemodel 转换为 Caffe2 pb 模型1. 单输入单输出 - caffe_translator.pyCaffe2 提供了将 caffemodel 转换为 caffe2 模型的工具——caffe_translator.py. 其使用:python -m caffe2.python.caffe_translator deploy.prototx原创 2018-01-03 09:56:41 · 8485 阅读 · 1 评论 -
Caffe2 - (八)图像加载与预处理
Caffe2 - 图像加载与预处理 举例说明从图像文件或图像 url 加载图像,以及相应的 Caffe2 用到的必要的图像预处理.必要的 python 包:sudo pip install scikit-image scipy matplotlibimport skimageimport skimage.io as ioimport skimage.transfo...原创 2018-01-05 15:31:36 · 3957 阅读 · 6 评论 -
Caffe2 - (九)MNIST 手写字体识别
Caffe2 - MNIST 手写字体识别 LeNet - CNN 网络训练; 采用 ReLUs 激活函数代替 Sigmoid. model helper import matplotlib.pyplot as pltimport numpy as npimport osimport shutilimport caffe2.python.predictor...原创 2018-01-06 14:42:31 · 3054 阅读 · 8 评论 -
Caffe2 - Detectron 安装
Detectron 安装Detectron 项目地址基于 Caffe2.Detectron 项目:Feature Pyramid Networks for Object DetectionMask R-CNN - Marr Prize at ICCV 2017Detecting and Recognizing Human-Object InteractionsFocal L...原创 2018-01-23 16:57:09 · 18887 阅读 · 28 评论 -
Caffe2 - Detectron 图片测试结果
Detectron 图片测试结果随机找的图片进行测试,结果很惊喜,很赞!!!1. Box 与 Maskpython2 tools/infer_simple.py \ --cfg configs/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml \ --output-dir /tmp/detectron-visua...原创 2018-01-24 11:23:43 · 4765 阅读 · 43 评论 -
Caffe2 - Detectron 初步使用
Detectron 初步使用Detectron 安装.Detectron 提供了基于 COCO Dataset 的推断和训练使用说明 - Using Detectron.1. Detectron Model ZOO 和 BaselinesDetectron 截止到 2017.12 提供了很多 baselines 模型,并记为 12_2017_baselines. Base...原创 2018-01-23 18:15:42 · 6897 阅读 · 9 评论 -
Caffe2 - (十)训练数据集创建
Caffe2 - 训练数据集创建caffe2 使用二值 DB 存储模型训练的数据,以 key-value 格式保存,key1 value1 key2 value2 key3 value3 ...DB 中,将 keys 和 values 保存为 strings 形式;可以通过 TensorProtos protocol buffer 来转换为结构化的数据: TensorPro...原创 2018-01-08 11:43:24 · 2224 阅读 · 0 评论 -
Caffe2 - (十三) 基于 Python 创建 Operator
Caffe2 - Python Operator类似于 Caffe 基于 Python 定制 CaffeLayers, Caffe2 也提供了使用 Python 来自定义 Caffe2 Operators.Forward Python OperatorCaffe2 提供了 high-level 接口,用于 Python ops 创建 —— Net.Python() 接口.原创 2018-01-25 10:59:37 · 2215 阅读 · 0 评论 -
Caffe2 - (四) 基于 squeezenet 分类的模型测试
Caffe2 模型加载与测试Model Zoo这里以 squeezenet 模型为例,对图片中的 object 分类.下载训练好的模型:python -m caffe2.python.models.download -i squeezenet模型加载:读取 protobuf 文件:with open("init_net.pb") as f: init_net原创 2017-12-29 11:35:39 · 3666 阅读 · 0 评论 -
Caffe2 - (十四) 网络构建API之 brew
Caffe2 - Brew Modelsbrew 是 Caffe2 用于构建模型的新的 API. 之前是 CNNModelHelper来构建模型. 但 Caffe2 不仅擅长 CNNs,还能够提供更加通用的ModelHelperobject.新的 ModelHelper与CNNModelHelper.brew具有很多相同的功能,使得其易于构建模型.原创 2018-01-25 22:27:52 · 1782 阅读 · 0 评论 -
Caffe2 - (二十七) Detectron 之 modeling - detector
Caffe2 - (二十七) Detectron 之 modeling - detectorDetecton 定义了一个 DetectionModelHelper 类,来表示 Detectron 模型.原创 2018-04-04 17:12:58 · 2028 阅读 · 2 评论