美文网首页
基于mxnet的Deep-Feature-Flow源码架构解析

基于mxnet的Deep-Feature-Flow源码架构解析

作者: 飞翔的小瓜瓜 | 来源:发表于2017-10-27 15:00 被阅读0次

    首先声明本人菜鸟一枚,如有任何错误欢迎指正!

    继做完Deep-Feature-Flow的PPT后,老大又让跑DFF的代码,但无奈之前专注于caffe框架,但DFF官方只给了mxnet框架上的源码(果然微软人家要推广自己家的框架哇),所以成为组里第一个转战mxnet的实习生【笑哭】。在跑通demo后(跑代码过程见mxnet编译运行DFF),后又花了大概3周的时间研读代码,构建起总的框架,期间收获了很多,也为下一步对DFF的改进打下了基础

    ----------------------------------------------------主子镇楼-----------------------------------------------------------

    一、mxnet下Deep-Feature-Flow的文件夹结构

    当打开下载的DFF包后,我们都是一脸萌币的。

    首先可以看一下READ_ME文件:包含了Introduction、Disclaimer、License、Requirements、Installation、Demo、Preparation for Training & Testing、Usage等部分内容,可以对整个DFF代码有一个初步的了解(当然是要在看完看透论文的基础上)。

    然后我们来看各个文件夹的内容

    DFF中主要文件夹

    data:包括了训练和测试使用的datasets相关文件

    demo:包括了在运行demo时使用到的输入图片,以及用来存储demo生成的标定结果图片。

    dff_rfcn:这个文件夹包括了DFF的绝大部分核心代码,包含神经网络的建立文件、用来进行训练和测试的function、class等。本文中,将对这一文件进行重点介绍。

    experiments:包括参数的配置文件(./experiments/{rfcn/dff_rfcn}/cfgs)、实验过程的日志文件以及train的源代码(引用了dff_rfcn中的各种函数)

    external:包括了与mxnet相关的文件(与我们的源代码基本无关~)

    lib:我们在各种安装包里都会见到的文件夹

    model:包括了训练所需的模型(ImageNet pre-trained ResNet-v1-101 model and Flying-Chairs pre-trained FlowNet model)

    output:用来存储训练及测试过程中的输出(A cache folder would be created automatically to save the model and the log under)

    rfcn:包含了rfcn网络的核心源代码

    二、各个文件的相互对应与import关系

    demo(简化版的experiments) 正式的train和test

    三、神经网络搭建文件解析

    见后续

    四、实验过程:训练与测试

    以正式训练时的dff_rfcn_end2end_train.test.py为例:

    dff_rfcn_end2end_train.test.py文件:

    dff_rfcn_end2end_train.test.py parse_arg函数

    其中train_net()函数是重点,源代码及注释见后续~

    相关文章

      网友评论

          本文标题:基于mxnet的Deep-Feature-Flow源码架构解析

          本文链接:https://www.haomeiwen.com/subject/kshbuxtx.html