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 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 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