https://github.com/aysenurcftc/mammography_images_analysis

https://github.com/aysenurcftc/mammography_images_analysis

Last synced: 2 days ago
JSON representation

• Host: GitHub
• URL: https://github.com/aysenurcftc/mammography_images_analysis
• Owner: aysenurcftc
• Created: 2024-06-12T17:01:49.000Z (5 months ago)
• Default Branch: main
• Last Pushed: 2024-07-15T17:49:25.000Z (4 months ago)
• Last Synced: 2024-07-15T21:53:59.046Z (4 months ago)
• Language: Python
• Size: 27.3 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        

# Breast Cancer Classification

## Installation

```bash

  pip install -r requirements.txt

```

    

## Usage/Examples

```javascript

usage: main.py [-h] [--dataset {CBIS-DDSM,CLAHE_images,MBCD_Implant}] [--mammogram_type MAMMOGRAM_TYPE] [--model {cnn,vit}] [--run_mode {training,test}]

               [--learning_rate LEARNING_RATE] [--batch_size BATCH_SIZE] [--max_epochs MAX_EPOCHS] [--verbose_mode VERBOSE_MODE] [--visualize VISUALIZE]

Training and testing a CNN model for different datasets.

optional arguments:

  -h, --help            show this help message and exit

  --dataset {CBIS-DDSM,CLAHE_images,MBCD_Implant}

                        Dataset to be used

  --mammogram_type MAMMOGRAM_TYPE

                        Type of mammogram

  --model {cnn,vit}     Model to be used

  --run_mode {training,test}

                        Run mode: training or test

  --learning_rate LEARNING_RATE

                        Learning rate for the model

  --batch_size BATCH_SIZE

                        Batch size for training

  --max_epochs MAX_EPOCHS

                        Maximum number of epochs

  --verbose_mode VERBOSE_MODE

                        Verbose mode

  --visualize VISUALIZE

                        Visualize dataset

```