FilesExpand file tree

 master

/

basic_usage.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

22 lines (16 loc) · 550 Bytes

 master

/

basic_usage.py

Copy path

Top

File metadata and controls

• Code
• Blame

22 lines (16 loc) · 550 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

#!/usr/bin/env python

from lobe import ImageModel

model = ImageModel.load('path/to/exported/model')

# Predict from an image file

result = model.predict_from_file('path/to/file.jpg')

# Predict from an image url

result = model.predict_from_url('http://path/to/file.jpg')

# Predict from Pillow image

from PIL import Image

img = Image.open('path/to/file.jpg')

result = model.predict(img)

# Print top prediction

print("Top prediction:", result.prediction)

# Print all classes

for label, prop in result.labels:

print(f"{label}: {prop*100:.6f}%")