Object Detection

SDK.detect_objects(self: tfsdk.SDK, tf_image: tfsdk.TFImage) → List[tfsdk.BoundingBox]

Detect people and objects in the image.

Parameters

tf_image - the input tfsdk.TFImage, returned by tfsdk.SDK.preprocess_image(). :Returns:

The a list of tfsdk.BoundingBox representing the detected objects.

static SDK.get_object_label_string(object_label: tfsdk.OBJECTLABEL) → str

Converts an object label to a string.

Parameters

object_label – the tfsdk.BoundingBox.label.

Returns

The string representation of the tfsdk.BoundingBox.label.

class tfsdk.BoundingBox

property height

The height of the bounding box.

property label

The object class as a label.

property probability

The detection’s confidence probability.

to_dict(self: tfsdk.BoundingBox) → dict

Return a dictionary representation of the object.

property top_left

The top left corner Point of the bounding box.

property width

The width of the bounding box.

class tfsdk.OBJECTLABEL

Members:

person

bicycle

car

motorcycle

airplane

bus

train

truck

boat

traffic light

fire hydrant

stop sign

parking meter

bench

bird

cat

dog

horse

sheep

cow

elephant

bear

zebra

giraffe

backpack

umbrella

handbag

tie

suitcase

frisbee

skis

snowboard

sports ball

kite

baseball bat

baseball glove

skateboard

surfboard

tennis racket

bottle

wine glass

cup

fork

knife

spoon

bowl

banana

apple

sandwich

orange

broccoli

carrot

hot dog

pizza

donut

cake

chair

couch

potted plant

bed

dining table

toilet

tv

laptop

mouse

remote

keyboard

cell phone

microwave

oven

toaster

sink

refrigerator

book

clock

vase

scissors

teddy bear

hair drier

toothbrush