Commit eca9f56b authored by Olegs NIKISINS's avatar Olegs NIKISINS
Browse files

Reverted changes in data_folder related to h5py import

parent 7762c4de
Pipeline #26241 passed with stage
in 9 minutes and 54 seconds
......@@ -22,7 +22,7 @@ from torchvision import transforms
import torch
# import h5py
import h5py
#==============================================================================
......@@ -55,8 +55,6 @@ def get_file_names_and_labels(files, data_folder, extension = ".hdf5", hldi_type
frame to extract the data from.
"""
import h5py
file_names_labels_keys = []
if hldi_type == "pad":
......@@ -202,8 +200,6 @@ class DataFolder(data.Dataset):
Index of the class.
"""
import h5py
path, target, key = self.file_names_labels_keys[index]
with h5py.File(path, "r") as f_h5py:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment