astroddpm icon indicating copy to clipboard operation
astroddpm copied to clipboard

train model on own data?

Open xiaoxiao079 opened this issue 1 year ago • 1 comments

hi there i want to train you model on our images. could it possible to train you model on grey scale images, and what are the changes which would needed to be done before training it on custom data?

xiaoxiao079 avatar Jan 23 '24 09:01 xiaoxiao079

Yes you can train on greyscale images, you would need to normalise them with a new dataloader so that they have a scale between -1 and 1. You would also need to change the network input channels to be 1.

Smith42 avatar May 01 '24 09:05 Smith42