Movatterモバイル変換


[0]ホーム

URL:


Skip to content

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up

A tensorflow implementation of FashionAI Global Challenge

License

NotificationsYou must be signed in to change notification settings

yd8534976/FashionAI-TF

Repository files navigation

FashionAI

Image data

DownloadImage data

Folder Structure

  • nets: store modified ResNet
  • model: store checkpoint files
  • outputs : store predicted files
  • summary: store files for tensorboard
  • train_set: place training data here
  • test_set: place test data here
  • fashion_evaluator.py: evaluator script
  • fashion_generator.py: data generator
  • fashion_helper.py: store a bunch of helper functions
  • fashion_stacked.py: main file to define model
  • train_script.py: train script
  • test_script.py: test script

Prerequisites

Docker is recommended:

  • nvidia-docker
  • pull image from Docker Hub
    docker pull yd8534976/tf-aiden

Alternatively:

  • python
  • tensorflow-gpu (>= 1.4)
  • numpy
  • pandas
  • opencv-python
  • jupyter (optional)
  • tensorboard (optional)

Pre-trained ResNet50 model

  • You can download pre-trained models from tensorflow officalslim model zoo.
  • Put checkpoint files intomodel/

Basic use

  • Download datasets and put them intotrain_set/
  • Download pre-trained ResNet-50 from slim model zoo
  • Configuretrain_script.py
  • Train end-to-end
    python train_script.py
  • Visualize your training using TensorBoard
    tensorboard --logdir=summary/
  • Generate predicted files
    python test_script.py
  • Visualize your prediction usingdemo_notes.ipynb
  • Evaluate your predictions
    python fashion_evaluator.py

Demo

  • blousedemo1
  • skirtdemo2
  • outweardemo3
  • dressdemo4
  • trousersdemo5

About

A tensorflow implementation of FashionAI Global Challenge

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

[8]ページ先頭

©2009-2025 Movatter.jp