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
|
``DropoutSSD``: wraps Dropout SSD 300 model
|
||||||
|
|
||||||
``SSD``: wraps vanilla 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 os
|
||||||
import pickle
|
import pickle
|
||||||
|
|||||||
Reference in New Issue
Block a user