Skip to content

WidgetA/sonar_baseline_with_tensorflow

Repository files navigation

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指定被推理图片放置的文件夹路径

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published