8 Star 11 Fork 5

明故为知 / FashionAI

加入 Gitee
与超过 800 万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README.md

FashionAI

Travis CircleCI License PRs Welcome

This repo is code of FashionAI Global Challenge—Attributes Recognition of Apparel based on PyTorch. This repo only for learning.

Environment

  • Operating system: Ubuntu 17.10
  • Data would take up to 25GB disk memory
  • Memory cost would be around 20GB
  • Dependencies:

Prerequisites

  • Download this repo

    git clone https://github.com/Lmy0217/FashionAI.git
    cd FashionAI
  • Install requirements

    pip3 install -r requirements.txt
  • (Unnecessary) Download the Attributes Recognition of Apparel dataset and extract the tar file in the folder datasets (now, this folder should contain three folder named 'base', 'web' and 'rank' respectively)

Usage

The training and testing scripts come with several options, which can be listed with the --help flag.

python3 main.py --help

To run the training and testing, simply run main.py. By default, the script runs resnet34 on attribute 'coat_length_labels' with 50 epochs.

To training and testing resnet34 on attribute 'collar_design_labels' with 100 epochs and some learning parameters:

python3 main.py --model 'resnet34' --attribute 'collar_design_labels' --epochs 100 --batch-size 128 --lr 0.01 --momentum 0.5

Every epoch trained model will be saved in the folder save/[attribute]/[model].

License

The code is licensed with the MIT license.

简介

FashionAI Global Challenge—Attributes Recognition of Apparel based on PyTorch. 展开 收起
Python
MIT
取消

发行版

暂无发行版

FashionAI

贡献者

全部

近期动态

加载更多
不能加载更多了
Python
1
https://gitee.com/lmy0217/FashionAI.git
git@gitee.com:lmy0217/FashionAI.git
lmy0217
FashionAI
FashionAI
master

搜索帮助