Commit 7762c4de authored by Olegs NIKISINS's avatar Olegs NIKISINS
Browse files

Added h5py to requirements

parent b2a8f933
Pipeline #26235 failed with stage
in 26 minutes and 4 seconds
......@@ -40,6 +40,9 @@ class ConvAutoencoder(nn.Module):
nn.Tanh())
def forward(self, x):
"""
The forward method.
"""
x = self.encoder(x)
x = self.decoder(x)
return x
......
......@@ -22,7 +22,7 @@ from torchvision import transforms
import torch
import h5py
# import h5py
#==============================================================================
......@@ -55,6 +55,8 @@ 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":
......@@ -200,6 +202,8 @@ 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:
......
......@@ -8,3 +8,4 @@ bob.io.base
bob.io.image
torch >= 0.4.0
torchvision >= 0.2.0
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