search

Zhengxinyang / LAS-Diffusion

MIT License
217 stars 15 forks source link

Missing file "white_image_feature.npy" #5

Closed Sharpiless closed 1 year ago

Sharpiless commented 1 year ago

I download ShapeNet and process the dataset with:

python prepare_sketch.py

I do get some generated folders in shapenet_edge:

angles  edge  feature

When I run bash scripts/train_sketch.sh, I get some errors:

  File "/share/liangyingping/Editing/LAS-Diffusion/network/model_trainer.py", line 118, in train_dataloader
    _dataset = occupancy_field_Dataset(sdf_folder=self.sdf_folder,
  File "/share/liangyingping/Editing/LAS-Diffusion/network/data_loader.py", line 79, in __init__
    self.white_image_feature = np.load(os.path.join(
  File "/data/liangyingping/anaconda3/envs/sketch_diffusion/lib/python3.9/site-packages/numpy/lib/npyio.py", line 405, in load
    fid = stack.enter_context(open(os_fspath(file), "rb"))
FileNotFoundError: [Errno 2] No such file or directory: 'data/ShapeNet/shapenet_edge/feature/white_image_feature.npy'

I have no idea where to generate "white_image_feature.npy". Are there any suggestions?

Zhengxinyang commented 1 year ago

You can copy white_image_feature.npy inside utils folder to data/ShapeNet/shapenet_edge/feature/

Sharpiless commented 1 year ago

Tips: There seems to be a typo in the code: https://github.com/Zhengxinyang/LAS-Diffusion/blob/main/sparsity_network/data_loader.py line 79:

_data_classes == [data_class]

seems to be changed to:

_data_classes = [data_class]
Zhengxinyang commented 1 year ago

Thanks, I will fix it!