Thank you very much, one more question, my neural network is not able to learn from my images, I am using around 4,560 images with ADHD and 5,200 with non-ADHD and I am getting this kind of results, any advice?
Epoch 0/99
----------
train Loss: 0.6730 Acc: 0.6294

Epoch 1/99
----------
train Loss: 0.6253 Acc: 0.6662

Epoch 2/99
----------
train Loss: 0.6153 Acc: 0.6746

Epoch 3/99
----------
train Loss: 0.6099 Acc: 0.6818

Epoch 4/99
----------
train Loss: 0.6031 Acc: 0.6880

Epoch 5/99
----------
train Loss: 0.5983 Acc: 0.6909

Epoch 6/99
----------
train Loss: 0.5950 Acc: 0.6956

Epoch 7/99
----------
train Loss: 0.5838 Acc: 0.7079

Epoch 8/99
----------
train Loss: 0.5779 Acc: 0.7073

Epoch 9/99
----------
train Loss: 0.5782 Acc: 0.7094

Epoch 10/99
----------
train Loss: 0.5763 Acc: 0.7123

Epoch 11/99
----------
train Loss: 0.5752 Acc: 0.7116

Epoch 12/99
----------
train Loss: 0.5751 Acc: 0.7163

Epoch 13/99
----------
train Loss: 0.5753 Acc: 0.7167

Epoch 14/99
----------
train Loss: 0.5754 Acc: 0.7128

Epoch 15/99
----------
train Loss: 0.5713 Acc: 0.7173

Epoch 16/99
----------
train Loss: 0.5761 Acc: 0.7116

Epoch 17/99
----------
train Loss: 0.5736 Acc: 0.7105

Epoch 18/99
----------
train Loss: 0.5721 Acc: 0.7141

Epoch 19/99
----------
train Loss: 0.5713 Acc: 0.7153

Epoch 20/99
----------
train Loss: 0.5736 Acc: 0.7137

Epoch 21/99
----------
train Loss: 0.5713 Acc: 0.7144

Epoch 22/99
----------
train Loss: 0.5713 Acc: 0.7164

Epoch 23/99
----------
train Loss: 0.5722 Acc: 0.7146

Epoch 24/99
----------
train Loss: 0.5734 Acc: 0.7145

Epoch 25/99
----------
train Loss: 0.5729 Acc: 0.7130

Epoch 26/99
----------
train Loss: 0.5723 Acc: 0.7173

Epoch 27/99
----------
train Loss: 0.5737 Acc: 0.7104

Epoch 28/99
----------



These are the images
ADHD
646ADHD24recurrence_plot998×998 35.8 KB
Control
418Control22recurrence_plot998×998 40.6 KB