Skip to content

Linardos/BFP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

86 Commits
 
 
 
 
 
 
 
 

Repository files navigation

BFP

Requirements

  • Docker
  • Host has NVIDIA GPUs
  • Data structure:
    • DATA_PATH/
      • [images_info].csv
      • images/

Installation

Option 1: Pulling image from dockerhub

docker pull registry.gitlab.bsc.es/eucanimage/fl_breast_mg_classification

Option 2: Building image

git clone https://github.com/Linardos/BFP.git
cd docker
docker build -t bfp_docker .

Run FD client

DATA_PATH=/home/lidia-garrucho/datasets/OPTIMAM/png_screening_cropped_fixed
CSV_FILENAME=client_images_screening.csv
IMAGES_FOLDER=images
DATA_LOADER_TYPE=optimam

cd BFP
docker run -it -v $DATA_PATH:/BFP/dataset -v $PWD/src:/BFP/src -e csv_path=/BFP/dataset/$CSV_FILENAME -e dataset_path=/BFP/dataset/$IMAGES_FOLDER -e data_loader_type=$DATA_LOADER_TYPE -e client_log_paths=/BFP/src/client_logs -e server=84.88.186.195:8080 bfp_docker


docker run -it -v $DATA_PATH:/BFP/dataset -v $PWD/src:/BFP/src -e csv_path=/BFP/dataset/$CSV_FILENAME -e data_loader_type=$DATA_LOADER_TYPE -e dataset_path=/BFP/dataset/images -e server=84.88.186.195:8080 -e client_log_paths=/BFP/src/client_logs -e NVIDIA_VISIBLE_DEVICES=2 --runtime=nvidia -e NVIDIA_DRIVER_CAPABILITIES=compute,utility registry.gitlab.bsc.es/eucanimage/fl_breast_mg_classification

For clients using BCDR or InBreast:

InBreast:

DATA_PATH=/home/lidia-garrucho/datasets/INBREAST
CSV_FILENAME=INbreast_updated_cropped_breast.csv
IMAGES_FOLDER=AllPNG_cropped
DATA_LOADER_TYPE=inbreast

docker run -it -v $DATA_PATH:/BFP/dataset -v $PWD/src:/BFP/src -e csv_path=/BFP/dataset/$CSV_FILENAME -e dataset_path=/BFP/dataset/images -e server=84.88.186.195:8080 bfp_docker

docker run -it -v $DATA_PATH:/BFP/dataset -v $PWD/src:/BFP/src -e csv_path=/BFP/dataset/$CSV_FILENAME -e data_loader_type=$DATA_LOADER_TYPE -e dataset_path=/BFP/dataset/AllPNG_cropped -e server=84.88.186.195:8080 -e client_log_paths=/BFP/src/client_logs -e NVIDIA_VISIBLE_DEVICES=2 --runtime=nvidia -e NVIDIA_DRIVER_CAPABILITIES=compute,utility registry.gitlab.bsc.es/eucanimage/fl_breast_mg_classification

BCDR:

DATA_PATH=/home/lidia-garrucho/datasets/BCDR
CSV_FILENAME=None
IMAGES_FOLDER=cropped
DATA_LOADER_TYPE=bcdr

docker run -it -v $DATA_PATH:/BFP/dataset -v $PWD/src:/BFP/src -e csv_path=/BFP/dataset/$CSV_FILENAME -e dataset_path=/BFP/dataset -e data_loader_type=$DATA_LOADER_TYPE -e server=84.88.186.195:8080 bfp_docker

docker run -it -v $DATA_PATH:/BFP/dataset -v $PWD/src:/BFP/src -e csv_path=/BFP/dataset/$CSV_FILENAME -e data_loader_type=$DATA_LOADER_TYPE -e dataset_path=/BFP/dataset/cropped -e server=84.88.186.195:8080 -e NVIDIA_VISIBLE_DEVICES=2 --runtime=nvidia -e NVIDIA_DRIVER_CAPABILITIES=compute,utility registry.gitlab.bsc.es/eucanimage/fl_breast_mg_classification

About

Breast cancer Federated Project

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published