Code Monkey home page Code Monkey logo

fashionai2018's Introduction

FashionAI2018 服装属性标签识别

环境

caffe+keras
python 2.7
numpy 1.14.2
opencv 3.4.0
cuda9.0
cudnn 7.0

文件说明:

caffe部分

使用InceptionV4举例,该网络在复赛中可以实现94.11的准确率。

  1. 训练网络使用training.sh

修改相应的路径,即Log、TOOLS、-weights后面的路径,其中,Inception-V4的pretrain-model请到caffe-model下载 。
使用方法bash training.sh [solver][Class][GPU]。
solver : solver文件路径,本代码中存放于solver文件夹
class: 类别名,用来命名log日志文件,即neck,collar...
GPU : GPU的型号,如0,1,2
2. caffe源码处理

caffe,并用文件中的src和include文件夹替换caffe中的src和include。src和include主要修改图片预处理部分,并开始支持多label输入。
3. 训练网络的solver文件夹

修改net:后面的路径,替换为相应的根目录。修改snapshot_prefix路径。
4. train文件夹

即相应的网络结构,修改训练文件的路径,在data/train/中
5. deployV4文件夹

测试用的网络结构。
6. test.py

12-crop测试使用的程序。修改里面对应的路径即可。修改好路径可以直接python test.py运行
7. tools文件夹

初赛和复赛使用的一些小程序,里面包含复赛使用的双模型融合程序,以及初赛使用的多模型融合程序,以及相应的多属性标签的制作程序。

:由于有8个属性,如果使用多任务训练,则拥有8个训练标签,不属于当前图片的label置-1,属于当前label的从0开始计数,具体请参考data/train/中的txt

Keras部分:

请参考培文大佬源码:
Keras代码
直接可用,效果低于InceptionV4,不过也不错。

检测部分:

切割的大图部分请参考MaskRCNN代码
小图请参考CPN代码

fashionai2018's People

Contributors

kaijin1995 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

fashionai2018's Issues

dataset

楼主您好,我对这个i任务特别感兴趣,也是真心的想学习您的优秀源码,可是苦于没有数据集。现在比赛结束官网上也不能下载了,我很想跑这个任务,您能否分享我一下数据集~

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.