Skip to content
Python Shell
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
pytoolkit @ 657e38f
submissions
.gitignore Initial commit. Oct 3, 2019
.gitmodules
LICENSE
README.md
_data.py
averaging.py
model_baseline.py
model_clean.py
predict.sh . Jan 9, 2020

README.md

浮世絵作者予測

https://prob.space/competitions/ukiyoe-author

データの配置

mkdir data
pushd data
wget https://probspace-stg.s3.amazonaws.com/problems/39/data/37/ukiyoe-train-labels.npz
wget https://probspace-stg.s3.amazonaws.com/problems/39/data/38/ukiyoe-test-imgs.npz
wget https://probspace-stg.s3.amazonaws.com/problems/39/data/39/ukiyoe-train-imgs.npz
popd

学習

./model_baseline.py
./model_mixup.py

推論

./predict.sh

Dockerを使うなら、

docker run --gpus=all --rm --interactive --tty --volume=$PWD:/usr/src/app keras-docker:0.1.0 ./predict.sh

models/averaging/submission.csv が最終結果。

動作確認済み環境

  • GTX 1080 ×2
  • Ubuntu 18.04.3 LTS
  • Python 3.7.4
  • albumentations==0.4.3
  • better-exceptions==0.2.2
  • numba==0.47.0
  • numpy==1.18.1
  • pandas==0.25.3
  • scikit-learn==0.22.1
  • scipy==1.4.1
  • tensorflow==2.1.0
  • tqdm==4.41.1

学習済みモデル

https://github.com/ak110/probspace_ukiyoe/releases/download/v1.0/20200112_probspace_ukiyoe.tar.bz2

You can’t perform that action at this time.