sjtuplayer / anomalydiffusion

[AAAI 2024] AnomalyDiffusion: Few-Shot Anomaly Image Generation with Diffusion Model
MIT License
114 stars 14 forks source link

If I need download any model from huggingface. #21

Open spencerzzzzzzz opened 4 months ago

sjtuplayer commented 3 months ago

No, we have provided all the models.

spencerzzzzzzz commented 3 months ago

但是我在运行main.py的时候遇到了这个问题,好像是因为网络问题。 Traceback (most recent call last): File "main.py", line 645, in model = load_model_from_config(config, opt.actual_resume) File "main.py", line 30, in load_model_from_config model = instantiate_from_config(config.model) File "/data/ZLP/anomalydiffusion-master/ldm/util.py", line 85, in instantiate_from_config return get_obj_from_str(config["target"])(config.get("params", dict()), kwargs) File "/data/ZLP/anomalydiffusion-master/ldm/models/diffusion/ddpm.py", line 477, in init self.instantiate_cond_stage(cond_stage_config) File "/data/ZLP/anomalydiffusion-master/ldm/models/diffusion/ddpm.py", line 573, in instantiate_cond_stage model = instantiate_from_config(config) File "/data/ZLP/anomalydiffusion-master/ldm/util.py", line 85, in instantiate_from_config return get_obj_from_str(config["target"])(config.get("params", dict()), kwargs) File "/data/ZLP/anomalydiffusion-master/ldm/util.py", line 93, in get_obj_from_str return getattr(importlib.import_module(module, package=None), cls) File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/importlib/init.py", line 127, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "", line 1014, in _gcd_import File "", line 991, in _find_and_load File "", line 975, in _find_and_load_unlocked File "", line 671, in _load_unlocked File "", line 783, in exec_module File "", line 219, in _call_with_frames_removed File "/data/ZLP/anomalydiffusion-master/ldm/modules/encoders/modules.py", line 7, in import kornia File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/kornia/init.py", line 10, in from . import ( File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/kornia/augmentation/init.py", line 43, in from .container import AugmentationSequential, ImageSequential, PatchSequential, VideoSequential File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/kornia/augmentation/container/init.py", line 1, in from .augment import AugmentationSequential File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/kornia/augmentation/container/augment.py", line 17, in from .patch import PatchSequential File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/kornia/augmentation/container/patch.py", line 9, in from kornia.contrib.extract_patches import extract_tensor_patches File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/kornia/contrib/init.py", line 4, in from .image_stitching import ImageStitcher File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/kornia/contrib/image_stitching.py", line 7, in from kornia.feature import LocalFeatureMatcher, LoFTR File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/kornia/feature/init.py", line 4, in from .integrated import ( File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/kornia/feature/integrated.py", line 52, in class LAFDescriptor(nn.Module): File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/kornia/feature/integrated.py", line 65, in LAFDescriptor patch_descriptor_module: nn.Module = HardNet(True), File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/kornia/feature/hardnet.py", line 66, in init pretrained_dict = torch.hub.load_state_dict_from_url( File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/torch/hub.py", line 731, in load_state_dict_from_url return torch.load(cached_file, map_location=map_location) File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/torch/serialization.py", line 713, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "/home/zlp/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/torch/serialization.py", line 938, in _legacy_load typed_storage._storage._set_from_file( RuntimeError: unexpected EOF, expected 197414 more bytes. The file might be corrupted.

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "main.py", line 888, in if trainer.global_rank == 0: NameError: name 'trainer' is not defined