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: 224224, 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 |