Skip to content

[Models: conf.pth] Wrong data path in conf.pth-files of the provided models #20

@GregorKobsik

Description

@GregorKobsik

Description:

The provided conf.pth-files seem inconsistent with the provided dataset (cmp. Example below). The dataloader PartGraphShapesDataset uses the provided argument data_path for self.data_dir. Later, in the function __getitem__(...) the geometric data geo_data is loaded given the provided data path. Unfortunately, all .npz-files are located in Chair_dgeo directory and not in the Chair_dhier directory.

Is my assumption correct, that we need to manually update the configuration files to make it consistent with the provided dataset (or vice versa)?

Example (exp_ae_Chair):

{'exp_name': 'PartNet_ae_Chair_4000_ccc',
 'category': 'Chair',
 'device': 'cuda:0',
 'seed': 3124256514,
 'model_path': '../data/models',
 'data_path': '../data/partnetdata/Chair_dhier',     <-- inconsistent with dataset, no ".npz"-files
 'floor_data_path': '',
 'train_dataset': 'chair.txt',
...
 'max_part_per_parent': 10,
 'X': 12,
 'Y': 2}

DSG-Net/code/datav1.py

Lines 464 to 465 in 6c6d117

geo_fn = os.path.join(self.data_dir, self.pg_shapes[index][idx] + '.npz')
geo_data = np.load(geo_fn)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions