Image2Image Neural Network (Deep Learning)

1 次查看(过去 30 天)
Hello I would like to implement a image2image NN. During training I would like to feed my network 2 images (Before & After) so that it can learn the manipulation (i.e. Denoising, filtering, correction, etc..).
I have built my network and tested it on a small dataset (100 images, each 64x64). Currently when training my network I must load in (ALL) the data into memory.
Does the trainNetwork fcn have the ability to handle a tbl input where the first column is a list of filepaths to my before files and the second column is a list of filepaths to my after files?
if not - what are my other options in terms of passing though a list of filenames for before/after images into my network?
These images must be TIFF to ensure 32bit files

回答(1 个)

Johanna Pingel
Johanna Pingel 2019-2-6
Try following this example:
The idea is to use mini-batch datastore to create the before and after images, which should handle the memory management for you.
Let me know if this doesn't solve your problem.

类别

Help CenterFile Exchange 中查找有关 Image Data Workflows 的更多信息

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by