Skip to content

Codes for the challenge of google ai open image

Notifications You must be signed in to change notification settings

tony32769/GoogleAIOpenImg2018

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 91.1%
  • Jupyter Notebook 8.7%
  • Other 0.2%