x-cold / cnn-fabric-classification

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

cnn-fabric-classification

训练

  • 1 数据集准备:将数据集按照文件夹分类(类别使用数字)
  • 2 修改配置文件config.py 配置好分类总数, 数据集路径等
  • 3 开启训练 python train.py

预测

  • 1 修改配置文件config.py 指定要加载的模型,测试集路径等
  • 2 开启预测 python predict.py

tensorboard 查看

  • 1 tensorboard --logdir=runs

About

License:MIT License


Languages

Language:Python 100.0%