basiclab / DA-OD-MEAA-PyTorch

Domain-Adaptive Object Detection via Uncertainty-Aware Distribution Alignment, ACM MM 2020
https://dl.acm.org/doi/10.1145/3394171.3413553
MIT License
18 stars 1 forks source link

demo_global.py is not working #2

Open Shuntw6096 opened 3 years ago

Shuntw6096 commented 3 years ago
 # 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'

Thanks

theblackcat102 commented 3 years ago

@erictseng610 is this a module from faster_rcnn? If so can you include it into the repo?

Shuntw6096 commented 3 years ago

Hi, I found another version of the faster-rcnn https://github.com/harsh-99/SCL/tree/master/lib/model/faster_rcnn

Is this the same as the missing one? Thanks.

erictseng610 commented 3 years ago

We have not finished demo_global.py yet. I'll fix this bug in the following days