I am trying to train a neural network to simulate the frequency response of transmission through a physical structure. I have a dataset of images where each image of a structure has an associated transmission response that differs over frequency.
Rather than having the neural network output the transmission response for a single frequency, I would like to have a fully-connected layer with N outputs that correspond to the transmission response for N frequencies contained in the spectrum. My problem is that ImageFolder and DataFolder default to use the folder structure to assign a single label to each image. How do I assign multiple labels to each image (where the label is the transmission response across N frequencies)? Thanks!