Table 5 A comparison between most famous toolkits and libraries for training mammography
From: Deep convolutional neural networks for mammography: advances, challenges and applications
| Â | 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 |