示例#1
0
    def download(self):
        print("Downloading {} dataset.".format(self.name))
        url = self.url.format(self.name)
        download_file(url, self.path, self.name + ".zip")

        # Datasets are zipped in a folder: unpack them
        parent = self.path
        subfolder = osp.join(self.path, self.name)
        for filename in os.listdir(subfolder):
            shutil.move(osp.join(subfolder, filename),
                        osp.join(parent, filename))
        os.rmdir(subfolder)

        x, adj, y, mask_tr, mask_va, mask_te = preprocess_data(
            self.path, self.name)

        # Save pre-processed data
        npz_file = osp.join(self.path, self.name) + ".npz"
        adj = adj.tocoo()
        np.savez(
            npz_file,
            x=x,
            adj_data=adj.data,
            adj_row=adj.row,
            adj_col=adj.col,
            adj_shape=adj.shape,
            y=y,
            mask_tr=mask_tr,
            mask_va=mask_va,
            mask_te=mask_te,
        )
示例#2
0
    def download(self):
        print('Downloading {} dataset{}.'
              .format(self.name, ' (clean)' if self.clean else ''))
        url = '{}/{}.zip'.format(self.url_clean if self.clean else self.url, self.name)
        download_file(url, self.path, self.name + '.zip')

        # Datasets are zipped in a folder: unpack them
        parent = self.path
        subfolder = osp.join(self.path, self.name)
        for filename in os.listdir(subfolder):
            shutil.move(osp.join(subfolder, filename), osp.join(parent, filename))
        os.rmdir(subfolder)
示例#3
0
    def download(self):
        print("Downloading ModelNet{} dataset.".format(self.name))
        url = self.url[self.name]
        download_file(url, self.path, "ModelNet" + self.name + ".zip")

        # Datasets are zipped in a folder: unpack them
        parent = self.path
        subfolder = osp.join(self.path, "ModelNet" + self.name)
        for filename in os.listdir(subfolder):
            shutil.move(osp.join(subfolder, filename),
                        osp.join(parent, filename))
        os.rmdir(subfolder)
        shutil.rmtree(osp.join(self.path, "__MACOSX"), ignore_errors=True)
示例#4
0
 def download(self):
     print("Downloading ModelNet{} dataset.".format(self.name))
     url = self.url[self.name]
     download_file(url, self.path, "ModelNet" + self.name + ".zip")
     shutil.rmtree(osp.join(self.path, "__MACOSX"), ignore_errors=True)
示例#5
0
文件: modelnet.py 项目: zdqf/spektral
 def download(self):
     print('Downloading ModelNet{} dataset.'.format(self.name))
     url = self.url[self.name]
     download_file(url, self.path, 'ModelNet' + self.name + '.zip')
     shutil.rmtree(osp.join(self.path, '__MACOSX'), ignore_errors=True)