def __init__(self, classes, pretrained=False, class_agnostic=False):
self.model_path = 'data/pretrained_model/alexnet_mxnet.pth'
self.dout_base_model = 512
self.pretrained = pretrained
self.class_agnostic = class_agnostic
_fastRCNN.__init__(self, classes, class_agnostic)
def __init__(self,
classes,
num_layers=101,
pretrained=False,
class_agnostic=False):
self.model_path = 'data/pretrained_model/resnet101_caffe.pth'
self.dout_base_model = 1024
self.pretrained = pretrained
self.class_agnostic = class_agnostic
_fastRCNN.__init__(self, classes, class_agnostic)
def __init__(self,
classes,
pretrained=False,
class_agnostic=False,
pretrained_weight=''):
if pretrained_weight == '':
self.model_path = 'data/pretrained_model/vgg16_caffe.pth'
else:
self.model_path = pretrained_weight
self.dout_base_model = 512
self.pretrained = pretrained
self.class_agnostic = class_agnostic
_fastRCNN.__init__(self, classes, class_agnostic)