From: Defect detection of gear parts in virtual manufacturing
Layer | Input channels | Kernel size | Stride | Output channels | Number of parameters |
---|---|---|---|---|---|
Com1_Conv1(+ BN1) | N × 3 | 1 | 1 | N × 64 | 256 (+ 256) |
Com1_Conv2(+ BN2) | N × 3 | 2 | 1 | N × 64 | 448 (+ 256) |
Com1_Conv3(+ BN3) | N × 3 | 3 | 1 | N × 64 | 640 (+ 256) |
Com1_Conv4(+ BN1) | N × 195 | 1 | 1 | N × 64 | 12544 (+ 256) |
Com2_Conv1(+ BN1) | N × 64 | 1 | 1 | N × 128 | 8320 (+ 512) |
Com2_Conv2(+ BN2) | N × 64 | 2 | 1 | N × 128 | 16512 (+ 512) |
Com2_Conv3(+ BN3) | N × 64 | 3 | 1 | N × 128 | 24704 (+ 512) |
Com2_Conv4(+ BN1) | N × 448 | 1 | 1 | N × 128 | 57472 (+ 512) |
Dense1 | N × 579 | - | - | N × 128 | 74240 |
Dense2 | N × 128 | - | - | N × 64 | 8256 |
Dense3 | N × 64 | - | - | N × 5 | 325 |
Total (trainable + non-trainable) | 205253 + 1536 |