added support for ResNeXt models

cvmodelz/models/factory.py

@@ -5,6 +5,7 @@ import pyaml
 from chainer import links as L
 from chainercv2.models import inceptionv3 as cv2inceptionv3
 from chainercv2.models import inceptionresnetv1 as cv2inceptionresnetv1
+from chainercv2.models import resnext as cv2resnext
 from chainercv2.models import resnet as cv2resnet
 from chainercv.links.model import ssd
 from chainercv.links.model import faster_rcnn
@@ -35,6 +36,7 @@ class ModelFactory(abc.ABC):
 
 			cv2inceptionv3.inceptionv3,
 			cv2inceptionresnetv1.inceptionresnetv1,
+			cv2resnext.resnext50_32x4d,
 		),
 
 		cvmodelz=(

cvmodelz/models/wrapper.py

@@ -55,7 +55,7 @@ class ModelWrapper(BaseModel):
 				input_size=299,
 			))
 
-		elif self.model_name == "ResNet":
+		elif self.model_name in ["ResNet", "ResNeXt"]:
 			info.update(dict(
 				input_size=224,
 			))