https://github.com/goruck/detection-server

Object detection served over grpc using Google Edge TPU.
https://github.com/goruck/detection-server

detection-server google-edge-tpu grpc object-detection

Last synced: about 2 months ago
JSON representation

Object detection served over grpc using Google Edge TPU.

• Host: GitHub
• URL: https://github.com/goruck/detection-server
• Owner: goruck
• License: mit
• Created: 2020-05-28T03:42:53.000Z (over 4 years ago)
• Default Branch: master
• Last Pushed: 2023-07-05T21:02:41.000Z (about 1 year ago)
• Last Synced: 2024-02-14T21:38:06.708Z (7 months ago)
• Topics: detection-server, google-edge-tpu, grpc, object-detection
• Language: Python
• Size: 41 KB
• Stars: 2
• Watchers: 2
• Forks: 1
• Open Issues: 4
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# detection-server

This project implements object detection served over grpc using the Google Edge TPU. These detections are returned from the server in the form of the centroid coordinates of the object's bounding box and its label. 

This is part of the [radar-ml](https://github.com/goruck/radar-ml) project.

# Installation

Clone this directory and cd to it.

Install the required Python modules in [requirements.txt](./requirements.txt) and OpenCV per the instructions [here](https://github.com/goruck/smart-zoneminder/tree/master/tpu-servers#installation).

If you'd like to compile the Protocol Buffers from scratch, use the following command.

```bash

$ python3 -m grpc_tools.protoc -I. --python_out=. --grpc_python_out=. ./detection_server.proto

```

# Usage

See [detection_server.proto](./detection_server.proto) for the server's interface definitions. 

The following illustrates an example client calling the detection server's services.

```python

"""

Example client for detection_server.py.

Copyright (c) 2020 Lindo St. Angel

"""

import grpc

import detection_server_pb2

import detection_server_pb2_grpc

def get_camera_resolution(stub):

    request = detection_server_pb2.Empty()

    try:

        response = stub.GetCameraResolution(request)

        print('Camera resolution fetched.')

        return response

    except grpc.RpcError as err:

        print(err.details()) #pylint: disable=no-member

        print('{}, {}'.format(err.code().name, err.code().value)) #pylint: disable=no-member

        exit(1)

def get_detected_objects(stub):

    request = detection_server_pb2.DesiredLabels(labels=['person', 'dog', 'cat'])

    try:

        response = stub.GetDetectedObjects(request)

        print('Detected object(s) fetched.')

        return response

    except grpc.RpcError as err:

        print(err.details()) #pylint: disable=no-member

        print('{}, {}'.format(err.code().name, err.code().value)) #pylint: disable=no-member

        exit(1)

def run():

    with grpc.insecure_channel('localhost:50051') as channel:

        stub = detection_server_pb2_grpc.DetectionServerStub(channel)

        while True:

            res = get_camera_resolution(stub)

            print(res)

            res = get_detected_objects(stub)

            print(res)

if __name__ == '__main__':

    run()

```

# License

Everything here is licensed under the [MIT license](./LICENSE).

# Contact

For questions or comments about this project please contact the author goruck (Lindo St. Angel) at {lindostangel} AT {gmail} DOT {com}.

# Acknowledgements

This project inspired by and borrows from [OpenCV camera examples with Coral](https://github.com/google-coral/examples-camera/tree/master/opencv).