226wyj / You-Draw-I-Guess

A image classification game based on CIFAR-10.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

You-Draw-I-Guess

An image classification system based on CIFAR-10.

项目依赖:

  • pytorch
  • PIL
  • tqdm
  • argparse
  • flask
  • Bootstrap 4.6.0

项目目录:

  • data文件夹用于存放CIFAR-10数据

训练:

python ./main.py --do_train --vgg --epoch=200

测试:

python ./main.py --do_eval --vgg

预测:

python ./main.py --do_predict --vgg

About

A image classification game based on CIFAR-10.

License:MIT License


Languages

Language:Python 67.9%Language:HTML 31.8%Language:CSS 0.3%