sonar_baseline_with_tensorflow

项目说明

本项目用于和鲸平台比赛，实践了用户 Pumpkin 分享的 baseline

参考项目链接

1. Protocol Buffer
2. Tensorflow 开源模型
3. TF-slim

环境

• CUDA 10.0
• cuDNN 7.6.0.64
• Tensorflow 1.15.2
• lxml 4.5.0
• Pillow 7.0.0
• matplotlib 3.2.0
• Cython 0.29.15
• pycocotools 2.0

使用方法

1. 运行 tfrecord_generator.py, 采用 -path 参数传入大赛数据集的压缩包
2. 运行 model_train.py, 采用 -path 参数传入预训练模型的文件夹地址
3. 运行 inference.py, 采用 -step指定希望被用于推理的训练步数, -path指定被推理图片放置的文件夹路径