Added train and predict functions to module documentation

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-03-21 17:14:44 +01:00
parent 7ba3333e28
commit 976563526a

View File

@ -29,6 +29,10 @@ Classes:
``DropoutSSD``: wraps Dropout SSD 300 model
``SSD``: wraps vanilla SSD 300 model
Functions:
predict(...): runs trained SSD/DropoutSSD on a given data set
train(...): trains the SSD/DropoutSSD on a given data set
"""
import os
import pickle