Skip to content

FashionAI Global Challenge—Attributes Recognition of Apparel based on PyTorch

License

Notifications You must be signed in to change notification settings

Lmy0217/FashionAI

Repository files navigation

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.

About

FashionAI Global Challenge—Attributes Recognition of Apparel based on PyTorch

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages