Skip to main content
. 2024 Jul 5;14:15537. doi: 10.1038/s41598-024-66543-7

Table 2.

Details of implementation specifications.

Hardware/deep learning framework Training hyper-parameters Data augmentation Time (ms)/img

Tensorflow: 2.13.0, Keras: 2.13.1, Cuda: 12.4, NVIDIA A100 40GB GPU

Intel Core Silver 4316 CPU x86_64, 2.30 GHz 128 GM RAM

Img size: 224×224, Batch: 8

Optimizer: SGD, loss: categorical cross-entropy, learning rate: 0.007

Gaussian noise

Random flip, rotation: 20, scale: 0.20, translation: 0.20

Using ResNet50

Train: 15.4

Inference: 5.8