Skip to content

Instantly share code, notes, and snippets.

@jeanpat
Created April 20, 2017 17:30
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 1 You must be signed in to fork a gist
  • Save jeanpat/df407399def84d65fa3e1bb61a57c219 to your computer and use it in GitHub Desktop.
Save jeanpat/df407399def84d65fa3e1bb61a57c219 to your computer and use it in GitHub Desktop.
Try to load dataset in tf_unet
Display the source blob
Display the rendered blob
Raw
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@JunMa11
Copy link

JunMa11 commented May 6, 2017

Thank you for sharing.
I have a training data set with the following shape:
Train_image.shape = (512, 512, 200), and the corresponding label.shape = (512, 512, 200). Both the types are int16.

Would it be possible for you to tell me how to gathered them in a variable (named Train_data) that can be used in trainer.train function?
path = trainer.train(Train_data, output_path, training_iters=10, epochs=5)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment