Commit 32226a3a authored by Yipeng Hu's avatar Yipeng Hu

ref #6 working training added with simplified data set

parent f77c6f96
......@@ -15,7 +15,7 @@ else:
home_dir = os.path.expanduser('~')
elif os.name == 'posix':
home_dir = os.environ['HOME']
filename = os.path.join(home_dir, 'Scratch/data/protocol/normalised/protocol_sweep_class_subjects.h5')
filename = os.path.join(home_dir, 'Scratch/data/protocol/normalised/ultrasound_50frames.h5')
# generate 5 random subjects
......
......@@ -16,7 +16,7 @@ else:
home_dir = os.path.expanduser('~')
elif os.name == 'posix':
home_dir = os.environ['HOME']
filename = os.path.join(home_dir, 'Scratch/data/protocol/normalised/protocol_sweep_class_subjects.h5')
filename = os.path.join(home_dir, 'Scratch/data/protocol/normalised/ultrasound_50frames.h5')
frame_size = tf.keras.utils.HDF5Matrix(filename, '/frame_size').data.value
frame_size = [frame_size[0][0],frame_size[1][0]]
......
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