Classification of augmented mammogram samples into normal,benign or malignant using CNN.
Framework used : Tensorflow
ARCHITECTURE USED :
Conv layer : 32 5X5 filters
Max pool layer
Conv layer : 64 5X5 filters
Max pool layer
Dense layer with 1024 units
Dense layer with 3 units