MiaoRain / GoogleAIOpenImg2018

Codes for the challenge of google ai open image

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Google AI Open Images - Object Detection Track

https://www.kaggle.com/c/google-ai-open-images-object-detection-track

File description

  • Training model (Faster and SSD): train_frcnn.py
  • Training model (Yolo): train_yolo.py
  • Generate the submission csv: eval.py
  • Evaluate with yolo: yolo_eval_csv.py
  • Convert raw img to tfrecord: create_oid_tf_record.py
  • Visual check the submission: vis_infer_result.py
  • Visual check tfrecord file: check_tfrecord.py

Working folder description

  • script: the bash files to run all the app
  • config: config files needed for app
  • All other main files should placed in the root folder
  • web: a simple framework to run a server app that can recogization simgle image.

Data file description

  • tfrecord files:
  • box.csv: ground truth of box and classes info for each image
  • oid_object_detection_challenge_500_label_map.pbtxt: mapping between machine code of class and human readable name
  • faster_rcnn.config: config files for tfrecord, checkpoint

Download dataset

Install Code

  • sudo /home/chamo/.pyenv/versions/anaconda3-5.1.0/bin/protoc ./object_detection/protos/*.proto --python_out=.

Requirement

  • tensorflow 1.8
  • Other python package which can be eaily installed by pip

Members:

Links

About

Codes for the challenge of google ai open image


Languages

Language:Python 91.0%Language:Jupyter Notebook 8.7%Language:Shell 0.1%Language:HTML 0.1%