HRNet / HRNet-Semantic-Segmentation

The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semantic segmentation for HRNet. https://arxiv.org/abs/1908.07919
Other
3.13k stars 686 forks source link

RuntimeError: No such file pretrained_models/hrnetv2_w48_imagenet_pretrained.pth #174

Open bberry25 opened 4 years ago

bberry25 commented 4 years ago

I tried to train pretrained model with cityscapes dataset and already downloaded pretrained model for training. But I can't find the place for pretrained_models//hrnetv2_w48_imagenet_pretrained.pth. Where is the directory for this issue?

The error code is screened like below: File "/home/uc/PY/HRNetOCR/tools/train.py", line 322, in main() File "/home/uc/PY/HRNetOCR/tools/train.py", line 99, in main '.get_seg_model')(config) File "/home/uc/PY/HRNetOCR/tools/../lib/models/seg_hrnet_ocr.py", line 678, in get_seg_model model.init_weights(cfg.MODEL.PRETRAINED) File "/home/uc/PY/HRNetOCR/tools/../lib/models/seg_hrnet_ocr.py", line 673, in init_weights raise RuntimeError('No such file {}'.format(pretrained)) RuntimeError: No such file pretrained_models/hrnetv2_w48_imagenet_pretrained.pth Traceback (most recent call last): File "/home/uc/PY/HRNetOCR/tools/train.py", line 322, in main() File "/home/uc/PY/HRNetOCR/tools/train.py", line 99, in main '.get_seg_model')(config) File "/home/uc/PY/HRNetOCR/tools/../lib/models/seg_hrnet_ocr.py", line 678, in get_seg_model model.init_weights(cfg.MODEL.PRETRAINED) File "/home/uc/PY/HRNetOCR/tools/../lib/models/seg_hrnet_ocr.py", line 673, in init_weights raise RuntimeError('No such file {}'.format(pretrained)) RuntimeError: No such file pretrained_models/hrnetv2_w48_imagenet_pretrained.pth => init weights from normal distribution => init weights from normal distribution Traceback (most recent call last): File "/home/uc/PY/HRNetOCR/tools/train.py", line 322, in main() File "/home/uc/PY/HRNetOCR/tools/train.py", line 99, in main '.get_seg_model')(config) File "/home/uc/PY/HRNetOCR/tools/../lib/models/seg_hrnet_ocr.py", line 678, in get_seg_model model.init_weights(cfg.MODEL.PRETRAINED) File "/home/uc/PY/HRNetOCR/tools/../lib/models/seg_hrnet_ocr.py", line 673, in init_weights raise RuntimeError('No such file {}'.format(pretrained)) RuntimeError: No such file pretrained_models/hrnetv2_w48_imagenet_pretrained.pth Traceback (most recent call last): File "/home/uc/PY/HRNetOCR/tools/train.py", line 322, in main() File "/home/uc/PY/HRNetOCR/tools/train.py", line 99, in main '.get_seg_model')(config) File "/home/uc/PY/HRNetOCR/tools/../lib/models/seg_hrnet_ocr.py", line 678, in get_seg_model model.init_weights(cfg.MODEL.PRETRAINED) File "/home/uc/PY/HRNetOCR/tools/../lib/models/seg_hrnet_ocr.py", line 673, in init_weights raise RuntimeError('No such file {}'.format(pretrained)) RuntimeError: No such file pretrained_models/hrnetv2_w48_imagenet_pretrained.pth Traceback (most recent call last): File "/home/uc/anaconda3/envs/pt_cu100/lib/python3.7/runpy.py", line 193, in _run_module_as_main "main", mod_spec) File "/home/uc/anaconda3/envs/pt_cu100/lib/python3.7/runpy.py", line 85, in _run_code exec(code, run_globals) File "/home/uc/anaconda3/envs/pt_cu100/lib/python3.7/site-packages/torch/distributed/launch.py", line 235, in main() File "/home/uc/anaconda3/envs/pt_cu100/lib/python3.7/site-packages/torch/distributed/launch.py", line 231, in main cmd=process.args) subprocess.CalledProcessError: Command '['/home/uc/anaconda3/envs/pt_cu100/bin/python', '-u', '/home/uc/PY/HRNetOCR/tools/train.py', '--local_rank=0', '--cfg', '/home/uc/PY/HRNetOCR/experiments/cityscapes/seg_hrnet_ocr_w48_train_512x1024_sgd_lr1e-2_wd5e-4_bs_12_epoch484.yaml']' returned non-zero exit status 1.

ghost commented 4 years ago

Read README.md link