sushi_detector_dataset
收藏数据集概述
- 名称: Tensorflow Sushi Detection
- 内容: 包含寿司图像数据集及用于训练自定义目标检测模型的脚本。
- 目标类别:
- 中脂金枪鱼
- 三文鱼
- 斑点鲥鱼
数据准备
-
创建TFRecord:
docker run -it --volume
pwd:/tensorflow floydhub/tensorflow:1.4.0-py3_aws.14 /tensorflow/docker/create_tf_record.sh -
下载预训练模型:
pushd data pretrained_model=ssd_mobilenet_v1_coco_11_06_2017 curl -OL http://download.tensorflow.org/models/object_detection/${ssd_mobilenet_v1_coco_11_06_2017}.tar.gz tar -xzf ${pretrained_model}.tar.gz rm -rf ${pretrained_model}.tar.gz ${pretrained_model} popd
运行环境
- Docker容器:
-
训练:
docker run -it --volume
pwd:/tensorflow floydhub/tensorflow:1.4.0-py3_aws.14 /tensorflow/docker/train.sh -
评估:
docker run -it --volume
pwd:/tensorflow floydhub/tensorflow:1.4.0-py3_aws.14 /tensorflow/docker/eval.sh -
Tensorboard:
tensorboard --logdir=data open http://localhost:6006
-
导出训练模型:
docker run -it --volume
pwd:/tensorflow floydhub/tensorflow:1.4.0-py3_aws.14 /bin/bash
-
运行在Floydhub
-
设置:
- 创建Floydhub账户
- 安装Floyd CLI
- 创建项目
- 创建数据集
-
训练:
- 从数据集运行
- 从过去作业的输出运行
-
评估与Tensorboard:
floyd run --tensorboard --env tensorflow-1.4 --data junji/projects/sushi_detector/{job_id}/output:data "bash ./floyd/setup.sh && cp -R /data/* /output && nohup sh ./floyd/eval.sh" && tensorboard --logdir=/output
运行在Google Cloud
- 注意: 由于Google Cloud ML Engine不支持Tensorflow 1.4版本,以下步骤不适用。




