You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
935d380d36
* added tfrecord loader/dumper * add comment * remove unused import * used the latest version of tensorflow(1.12.3) which supports cuda 9.0 updated cudnn library install tensorflow by default and replace it by tensorflow-gpu in case of cuda support is enabled * Updated changelog |
7 years ago | |
|---|---|---|
| .. | ||
| README.md | 7 years ago | |
| docker-compose.tf_annotation.yml | 7 years ago | |
| install.sh | 7 years ago | |
README.md
Tensorflow Object Detector
What is it?
- This application allows you automatically to annotate many various objects on images.
- It uses Faster RCNN Inception Resnet v2 Atrous Coco Model from tensorflow detection model zoo
- It can work on CPU (with Tensorflow or OpenVINO) or GPU (with Tensorflow GPU).
- It supports next classes (just specify them in "labels" row):
'surfboard', 'car', 'skateboard', 'boat', 'clock',
'cat', 'cow', 'knife', 'apple', 'cup', 'tv',
'baseball_bat', 'book', 'suitcase', 'tennis_racket',
'stop_sign', 'couch', 'cell_phone', 'keyboard',
'cake', 'tie', 'frisbee', 'truck', 'fire_hydrant',
'snowboard', 'bed', 'vase', 'teddy_bear',
'toaster', 'wine_glass', 'traffic_light',
'broccoli', 'backpack', 'carrot', 'potted_plant',
'donut', 'umbrella', 'parking_meter', 'bottle',
'sandwich', 'motorcycle', 'bear', 'banana',
'person', 'scissors', 'elephant', 'dining_table',
'toothbrush', 'toilet', 'skis', 'bowl', 'sheep',
'refrigerator', 'oven', 'microwave', 'train',
'orange', 'mouse', 'laptop', 'bench', 'bicycle',
'fork', 'kite', 'zebra', 'baseball_glove', 'bus',
'spoon', 'horse', 'handbag', 'pizza', 'sports_ball',
'airplane', 'hair_drier', 'hot_dog', 'remote',
'sink', 'dog', 'bird', 'giraffe', 'chair'.
- Component adds "Run TF Annotation" button into dashboard.
Build docker image
# From project root directory
docker-compose -f docker-compose.yml -f components/tf_annotation/docker-compose.tf_annotation.yml build
Run docker container
# From project root directory
docker-compose -f docker-compose.yml -f components/tf_annotation/docker-compose.tf_annotation.yml up -d