https://github.com/076923/tensorflow-2.0-object-detection-api

Tensorflow 2.x Object Detection API to use easily. Tensorflow 2.x Object Detection API Tutorial
https://github.com/076923/tensorflow-2.0-object-detection-api

object-detection python tensorflow2 tensorflow2-object-detection tutorial

Last synced: about 1 month ago
JSON representation

Tensorflow 2.x Object Detection API to use easily. Tensorflow 2.x Object Detection API Tutorial

• Host: GitHub
• URL: https://github.com/076923/tensorflow-2.0-object-detection-api
• Owner: 076923
• License: apache-2.0
• Created: 2021-03-14T13:07:13.000Z (almost 5 years ago)
• Default Branch: master
• Last Pushed: 2021-03-14T13:12:35.000Z (almost 5 years ago)
• Last Synced: 2025-01-20T09:32:31.760Z (about 1 year ago)
• Topics: object-detection, python, tensorflow2, tensorflow2-object-detection, tutorial
• Language: Jupyter Notebook
• Homepage:
• Size: 1.63 MB
• Stars: 2
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# TensorFlow 2.0 Object Detection API

![logo](images/logo.png) 

1. 모델 파일을 자동으로 다운로드합니다.

2. OpenCV를 활용하여 이미지를 입력받습니다.

3. 검출된 `classes`, `scores`, `boxes`를 반환합니다.

4. 간단한 시각화를 지원합니다.




## Requirements

[![Python 3.8](https://img.shields.io/badge/Python-3.8-3776AB)](https://www.python.org/downloads/release/python-380/)

[![TensorFlow 2.3](https://img.shields.io/badge/TensorFlow-2.3-FF6F00?logo=tensorflow)](https://github.com/tensorflow/tensorflow/releases/tag/v2.3.0)

[![Protobuf Compiler >= 3.0](https://img.shields.io/badge/ProtoBuf%20Compiler-%3E3.0-brightgreen)](https://grpc.io/docs/protoc-installation/#install-using-a-package-manager)

[![OpenCV >= 4.0](https://img.shields.io/badge/OpenCV-%3E4.0-FF6F00?logo=opencv)](https://github.com/tensorflow/tensorflow/releases/tag/v2.2.0)

[![tpqm](https://img.shields.io/badge/tpqm-4.59.0-brightgreen)](https://grpc.io/docs/protoc-installation/#install-using-a-package-manager)




## Installation

Clone the TensorFlow 2.0 Object Detection API repository.

    git clone https://github.com/076923/Tensorflow-2.0-Object-Detection-API.git




## Using

```python3

# Using GPU computing

import tensorflow as tf

physical_device = tf.config.list_physical_devices('GPU')

tf.config.experimental.set_memory_growth(physical_device[0], enable=True)

# Model Prediction

from core.detection import ModelZoo

model = ModelZoo(ModelZoo.SSD_MobileNet_v2_320x320)

img, input_tensor = model.load_image('images/dog.jpg')

classes, scores, boxes = model.predict(input_tensor)

visual = model.visualization(img, classes, scores, boxes, 0.7)

# OpenCV Visualization

import cv2

cv2.imshow("visual", visual)

cv2.waitKey()

```

| src | dst |

|:---:|:---:|

| ![src](images/dog.jpg) | ![dst](images/visual.png) |







![gpu](images/Using-GPU-computing.png)

![prediction](images/Model-Prediction.png)

![visualization](images/Visualization.png)




## Model List

- ModelZoo.CenterNet_Resnet50_V1_FPN_512x512

- ModelZoo.CenterNet_Resnet101_V1_FPN_512x512

- ModelZoo.CenterNet_Resnet50_V2_512x512

- ModelZoo.CenterNet_MobileNetV2_FPN_512x512

- ModelZoo.EfficientDet_D0_512x512

- ModelZoo.EfficientDet_D1_640x640

- ModelZoo.EfficientDet_D2_768x768

- ModelZoo.EfficientDet_D3_896x896

- ModelZoo.EfficientDet_D4_1024x1024

- ModelZoo.EfficientDet_D5_1280x1280

- ModelZoo.EfficientDet_D6_1280x1280

- ModelZoo.EfficientDet_D7_1536x1536

- ModelZoo.SSD_MobileNet_v2_320x320

- ModelZoo.SSD_MobileNet_V1_FPN_640x640

- ModelZoo.SSD_MobileNet_V2_FPNLite_320x320

- ModelZoo.SSD_ResNet50_V1_FPN_640x640_RetinaNet50

- ModelZoo.SSD_ResNet50_V1_FPN_1024x1024_RetinaNet50	

- ModelZoo.SSD_ResNet101_V1_FPN_640x640_RetinaNet101

- ModelZoo.SSD_ResNet101_V1_FPN_1024x1024_RetinaNet101

- ModelZoo.SSD_ResNet152_V1_FPN_640x640_RetinaNet152	

- ModelZoo.SSD_ResNet152_V1_FPN_1024x1024_RetinaNet152	

- ModelZoo.Faster_RCNN_ResNet50_V1_640x640

- ModelZoo.Faster_RCNN_ResNet50_V1_1024x1024

- ModelZoo.Faster_RCNN_ResNet50_V1_800x1333

- ModelZoo.Faster_RCNN_ResNet101_V1_640x640

- ModelZoo.Faster_RCNN_ResNet101_V1_1024x1024

- ModelZoo.Faster_RCNN_ResNet101_V1_800x1333

- ModelZoo.Faster_RCNN_ResNet152_V1_640x640

- ModelZoo.Faster_RCNN_ResNet152_V1_1024x1024

- ModelZoo.Faster_RCNN_ResNet152_V1_800x1333




## Test Models

- ModelZoo.CenterNet_Resnet101_V1_FPN_512x512

- ModelZoo.SSD_MobileNet_v2_320x320

- ModelZoo.SSD_ResNet50_V1_FPN_640x640_RetinaNet50




## Model Zoo

We provide a large collection of models that are trained on COCO 2017 in the Model Zoo.




## Authors

윤대희(s076923@gmail.com)




## License

Apache License 2.0