# initilize the network here.
from model.faster_rcnn.vgg16_SCL import vgg16
if args.net == 'vgg16':
fasterRCNN = vgg16(foggy_classes, pretrained=True)
# fasterRCNN = vgg16(pascal_classes, pretrained=False, class_agnostic=args.class_agnostic,lc=args.lc,gc=args.gc)
elif args.net == 'res101':
fasterRCNN = resnet(pascal_classes, 101, pretrained=False, class_agnostic=args.class_agnostic,context=args.context)
elif args.net == 'res50':
fasterRCNN = resnet(pascal_classes, 50, pretrained=False, class_agnostic=args.class_agnostic)
elif args.net == 'res152':
fasterRCNN = resnet(pascal_classes, 152, pretrained=False, class_agnostic=args.class_agnostic)
else:
print("network is not defined")
pdb.set_trace()
fasterRCNN.create_architecture()
It seems that some files are missing.
Traceback (most recent call last):
File "demo_global.py", line 192, in <module>
from model.faster_rcnn.vgg16_SCL import vgg16
ModuleNotFoundError: No module named 'model.faster_rcnn.vgg16_SCL'