|
|
@@ -5,6 +5,8 @@ import pyaml
|
|
|
from chainer import links as L
|
|
|
from chainercv2.models import inceptionv3 as cv2inceptionv3
|
|
|
from chainercv2.models import resnet as cv2resnet
|
|
|
+from chainercv.links.model import ssd
|
|
|
+from chainercv.links.model import faster_rcnn
|
|
|
from collections import OrderedDict
|
|
|
|
|
|
from cvmodelz.models import pretrained
|
|
|
@@ -22,7 +24,8 @@ class ModelFactory(abc.ABC):
|
|
|
),
|
|
|
|
|
|
chainercv=(
|
|
|
- # todo: chainercv.links.models.ssd
|
|
|
+ ssd.SSD300,
|
|
|
+ faster_rcnn.FasterRCNNVGG16,
|
|
|
),
|
|
|
|
|
|
chainercv2=(
|
