sjtuplayer / anomalydiffusion

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

What is "Personalized_mvtec_mask" ? #31

Open ayantian opened 4 months ago

ayantian commented 4 months ago

Traceback (most recent call last): File "train_mask.py", line 840, in data.prepare_data() File "/root/anaconda3/envs/Anomalydiffusion/lib/python3.8/site-packages/pytorch_lightning/core/datamodule.py", line 474,in wrapped_fn fn(*args, kwargs) File "/ssd2/cly/project/anomalydiffusion/main.py", line 298, in prepare_data instantiate_from_config(data_cfg) File "/ssd2/cly/project/anomalydiffusion/ldm/util.py", line 86, in instantiate_from_config return get_obj_from_str(config["target"])(config.get("params", dict()), **kwargs) File "/ssd2/cly/project/anomalydiffusion/ldm/util.py", line 94, in get_obj_from_str return getattr(importlib.import_module(module, package=None), cls) AttributeError: module 'ldm.data.personalized' has no attribute 'Personalized_mvtec_mask'