Train a deep learning model with filename in single csv file.

Question

Pratham Shah 2023-3-29

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1937294-train-a-deep-learning-model-with-filename-in-single-csv-file

编辑： Pratham Shah 2023-5-1

I want to train U-Net model for an application. Images are having multiple bands saved in different .tif file. To train DL model in MATLAB we require an imagedatastore. But I have filenames for all the images saved in single .csv file, so I'm not able to make a custom 'matReader' function for it.

I have done this when one .csv file contains filenames of different band and mask for a 'single image'. But in this approach I had to make multiple csv files. I want to do the process from a single csv file.

Structure of .csv file is something like:

Image1_B1, Image1_B2...., Mask1

Image2_B1, Image2_B2..., Mask2

.

ImageN_B1, ImageN_B2..., MaskN

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Abhijeet 2023-4-3

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1937294-train-a-deep-learning-model-with-filename-in-single-csv-file#answer_1207369

在 MATLAB Online 中打开

Hi,

You can create a custom matReader function to read the image data from the filenames stored in the csv file, by the following approach:

% Read the csv file into the fileList variable
fileList = readtable('image_file_list.csv');
% The customMatReader function to read the files from the filename variable
function img = customMatReader(filename, imageSize)
    img = imread(filename);
    img = imresize(img, imageSize);
end
% The size of the images, you should modify it as per your images
imageSize = [256, 256];
% Split the columns in the CSV into separate variables
fileList = splitvars(fileList);
% Create an imageDatastore from the filenames in the CSV file.
% The 'ReadFcn' option specifies a custom function to read the image data from each filename. 
% The resulting 'imds' object can be used for training the U-Net model.
imds = imageDatastore(fileList{:,1:end-1}, 'ReadFcn', @(filename) customMatReader(filename, imageSize));

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Pratham Shah 2023-4-5

在 MATLAB Online 中打开

Hi,

Thank you for your response!

I would like to clarify that one row consists of different channels of a single image. So, I need to pass one entire row of csv file in an instance. Also, as I'm reading csv using,

fileList = readtable('image_file_list.csv','Delimiter',',');

This line isn't making any difference

fileList = splitvars(fileList);

I have custom read function which will make a single image by stacking all the channels. Please tell me how can I pass single row to imageDatastore at a time.

请先登录，再进行评论。