Skip to content

Conversation

@iamrinni
Copy link

@iamrinni iamrinni commented Aug 2, 2021

Initial part for Google Coral facenet module.
Function usage example to create embeddings:
emb_calculator = Calculator()

for 1 image

emb_calculator.calc_embedding(img, mode='CPU') or 
emb_calculator.calc_embedding(img, resize=True,mode='TPU')

for several images

emb_calculator._calculate_embeddings(images, mode='CPU') or 
emb_calculator._calculate_embeddings(images, resize=True, mode='TPU')

To run docker image with TPU access: docker run -it --privileged --device=/dev/bus/usb:/dev/bus/usb test_coral /bin/bash

@iamrinni iamrinni requested a review from pospielov August 2, 2021 09:11
@github-actions
Copy link
Contributor

github-actions bot commented Aug 2, 2021

CLA Assistant Lite bot All contributors have signed the CLA ✍️

@ptichalouf
Copy link

any news about CORAL TPU support ?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants