tryolabs / luminoth

Deep Learning toolkit for Computer Vision.
https://tryolabs.com
BSD 3-Clause "New" or "Revised" License
2.4k stars 399 forks source link

Overhaul of the Python API #205

Closed nagitsu closed 6 years ago

nagitsu commented 6 years ago

Adds several utility functions to ease the use of Luminoth through Python:

With this, the following is now possible:

from luminoth import Detector, read_image, vis_objects

image = read_image('cat.png')

# Assumes `accurate` checkpoint by default. Other checkpoint, or a config object,
# can be specified.           
detector = Detector()
objects = detector.predict(image)

print(objects)

vis_objects(image, objects).save('cat-out.png')

Further work is required along the following lines: