IDEA-Research / DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"
Apache License 2.0
2.1k stars 230 forks source link

Code trying to download backbone from the internet? #187

Open htcml opened 1 year ago

htcml commented 1 year ago

My server has no internet access. I have download checkpoint and set up the environment.

When I run this command:

bash scripts/DINO_eval.sh /path/to/your/COCODIR /path/to/your/checkpoint

I got this errors:

raceback (most recent call last):
  File "main.py", line 388, in <module>
    main(args)
  File "main.py", line 143, in main
    model, criterion, postprocessors = build_model_main(args)
  File "main.py", line 83, in build_model_main
    model, criterion, postprocessors = build_func(args)
  File "/home/thuan/doc_ai/signature/DINO/models/dino/dino.py", line 713, in build_dino
    backbone = build_backbone(args)
  File "/home/thuan/doc_ai/signature/DINO/models/dino/backbone.py", line 170, in build_backbone
    batch_norm=FrozenBatchNorm2d)
  File "/home/thuan/doc_ai/signature/DINO/models/dino/backbone.py", line 120, in __init__
    pretrained=is_main_process(), norm_layer=batch_norm)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/site-packages/torchvision/models/_utils.py", line 142, in wrapper
    return fn(*args, **kwargs)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/site-packages/torchvision/models/_utils.py", line 228, in inner_wrapper
    return builder(*args, **kwargs)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/site-packages/torchvision/models/resnet.py", line 729, in resnet50
    return _resnet(Bottleneck, [3, 4, 6, 3], weights, progress, **kwargs)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/site-packages/torchvision/models/resnet.py", line 301, in _resnet
    model.load_state_dict(weights.get_state_dict(progress=progress))
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/site-packages/torchvision/models/_api.py", line 66, in get_state_dict
    return load_state_dict_from_url(self.url, progress=progress)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/site-packages/torch/hub.py", line 731, in load_state_dict_from_url
    download_url_to_file(url, cached_file, hash_prefix, progress=progress)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/site-packages/torch/hub.py", line 597, in download_url_to_file
    u = urlopen(req)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/urllib/request.py", line 222, in urlopen
    return opener.open(url, data, timeout)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/urllib/request.py", line 525, in open
    response = self._open(req, data)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/urllib/request.py", line 543, in _open
    '_open', req)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/urllib/request.py", line 503, in _call_chain
    result = func(*args)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/urllib/request.py", line 1393, in https_open
    context=self._context, check_hostname=self._check_hostname)
  File "/axp/aida/data/platformds/aiservices/conda/envs/donut/lib/python3.7/urllib/request.py", line 1352, in do_open
    raise URLError(err)
urllib.error.URLError: <urlopen error [Errno 97] Address family not supported by protocol>

Does the code try to download the backbone(in this case Res50) from the hub? If so, is there a way to download this backbone and run the code offline?