Added train and predict functions to module documentation
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -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
|
||||
|
||||
Reference in New Issue
Block a user