Table 5.
A comparison between most famous toolkits and libraries for training mammography
| Interface | Languages | Open source | CUDA support | Pre-trained models | Forks (Github) | Contributions (Github) | |
|---|---|---|---|---|---|---|---|
| TensorFlow | Python | C++, Python | Yes | Yes | Yes | 63,603 | 1,481 |
| Keras | Python, R | Python | Yes | Yes | Yes | 11,203 | 681 |
| Cafee | Python, Matlab, C++ | C++, Python | Yes | Yes | Yes | 14,868 | 267 |
| PyTorch | Python | C, Python, CUDA | Yes | Yes | Yes | 3,592 | 644 |
| MatConvNet | Matlab | CUDA | Yes | Yes | Yes | 651 | 24 |