├── .gitignore ├── LICENSE ├── README.md ├── algorithm ├── fedavg │ ├── client.py │ ├── fedavg_trainer.py │ ├── optim.py │ └── util.py ├── feddc │ ├── client.py │ ├── feddc_trainer.py │ └── util.py ├── feddyn │ ├── client.py │ ├── comm_helpers.py │ ├── feddyn_trainer.py │ └── optim.py ├── fedprox │ ├── client.py │ ├── comm_helpers.py │ ├── fedprox_trainer.py │ └── optim.py └── local │ ├── client.py │ ├── local_trainer.py │ ├── optim.py │ └── util.py ├── code_structure.md ├── core ├── base_dataset.py ├── base_model.py ├── base_network.py ├── logger.py ├── praser.py └── util.py ├── data ├── __init__.py ├── data_processing │ └── data_partion.py ├── dataloader │ └── data_loader.py └── dataset │ ├── __init__.py │ ├── aligned_dataset.py │ ├── base_dataset.py │ ├── image_folder.py │ └── utils.py ├── ddpm ├── LICENSE ├── README.md ├── README.md.backup ├── README_cn.md ├── config │ ├── AAF_config.7z │ ├── AAFnew0_train.json │ ├── CelebaHQ_face_train.json │ ├── labeltoimage_face0.json │ ├── labeltoimage_face1.json │ ├── labeltoimage_face2.json │ ├── labeltoimage_material0.json │ ├── labeltoimage_material0_seperate.json │ ├── labeltoimage_material0_seperate_test.json │ ├── labeltoimage_material1.json │ ├── labeltoimage_material1_seperate.json │ ├── labeltoimage_material1_seperate_test.json │ ├── labeltoimage_medical0.json │ ├── labeltoimage_medical0_material1.json │ └── labeltoimage_medical1.json ├── core │ ├── base_dataset.py │ ├── base_model.py │ ├── base_network.py │ ├── logger.py │ ├── praser.py │ └── util.py ├── data │ ├── __init__.py │ ├── dataset.py │ └── util │ │ ├── auto_augment.py │ │ └── mask.py ├── eval.py ├── misc │ ├── Palette Image-to-Image Diffusion Models.pdf │ └── image │ │ ├── Mask_Places365_test_00143399.jpg │ │ ├── Mask_Places365_test_00144085.jpg │ │ ├── Mask_Places365_test_00209019.jpg │ │ ├── Mask_Places365_test_00263905.jpg │ │ ├── Out_Places365_test_00143399.jpg │ │ ├── Out_Places365_test_00144085.jpg │ │ ├── Out_Places365_test_00209019.jpg │ │ ├── Out_Places365_test_00263905.jpg │ │ ├── Process_02323.jpg │ │ ├── Process_26190.jpg │ │ ├── Process_Places365_test_00042384.jpg │ │ └── Process_Places365_test_00309553.jpg ├── models.join │ ├── __init__.py │ ├── guided_diffusion_modules │ │ ├── nn.py │ │ └── unet.py │ ├── loss.py │ ├── metric.py │ ├── model.py │ ├── network.py │ └── sr3_modules │ │ └── unet.py ├── models │ ├── __init__.py │ ├── cnn_modules │ │ ├── __init__.py │ │ └── unet.py │ ├── guided_diffusion_modules │ │ ├── nn.py │ │ └── unet.py │ ├── loss.py │ ├── metric.py │ ├── model.py │ ├── network.py │ └── sr3_modules │ │ └── unet.py ├── preprocess │ └── mirflickr25k_preprocess.py ├── requirements.txt ├── run.py ├── scripts │ └── run_AAF.sh ├── slurm │ └── inpainting_places2.slurm ├── test.sh ├── tools │ ├── change_image_format.py │ ├── copy_fake_image_to_target_dir.py │ ├── copy_fake_image_to_target_dir_join.py │ ├── create_config_for_diffusion_seperate.py │ ├── create_flist.py │ ├── create_test_config_for_diffusion_join.py │ ├── create_test_config_for_diffusion_seperate.py │ ├── data_counter.py │ ├── data_merge.py │ ├── data_partition.py │ ├── data_select.py │ ├── delete_file.py │ ├── delete_pkl.py │ ├── draw_face_color.py │ ├── get_pred_evaluation.py │ ├── get_style_images.py │ ├── image_collection.py │ ├── mask_id_map.py │ ├── multi_dir_to_single.py │ ├── predict.py │ ├── predict_face_seg.py │ └── t-SNE_test.py └── train.sh ├── docs ├── Abstract.png └── Network.png ├── models ├── __init__.py ├── fedst_ddpm_model │ ├── base_model.py │ ├── ddpm_models │ │ ├── config │ │ │ ├── AAFnew0_join_train.json │ │ │ ├── colorization_mirflickr25k.json │ │ │ ├── inpainting_celebahq.json │ │ │ ├── inpainting_places2.json │ │ │ ├── labeltoimage.json │ │ │ ├── labeltoimage_aaf_face_new.json │ │ │ ├── labeltoimage_face0.json │ │ │ ├── labeltoimage_face1.json │ │ │ ├── labeltoimage_face2.json │ │ │ ├── labeltoimage_material0.json │ │ │ ├── labeltoimage_material1.json │ │ │ ├── labeltoimage_medical0.json │ │ │ ├── labeltoimage_medical1.json │ │ │ └── uncropping_places2.json │ │ ├── core │ │ │ ├── base_dataset.py │ │ │ ├── base_model.py │ │ │ ├── base_network.py │ │ │ ├── logger.py │ │ │ ├── praser.py │ │ │ └── util.py │ │ ├── data │ │ │ ├── __init__.py │ │ │ ├── dataset.py │ │ │ └── util │ │ │ │ ├── auto_augment.py │ │ │ │ └── mask.py │ │ ├── ddpm_config.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── guided_diffusion_modules │ │ │ │ ├── nn.py │ │ │ │ └── unet.py │ │ │ ├── loss.py │ │ │ ├── metric.py │ │ │ ├── model.py │ │ │ ├── network.py │ │ │ └── sr3_modules │ │ │ │ └── unet.py │ │ └── network.py │ ├── fedst_ddpm_model.py │ └── networks.py ├── losses │ ├── __init__.py │ └── losses.py └── unet_model │ ├── base_model.py │ ├── networks.py │ └── unet_model.py ├── opt ├── __init__.py ├── aaf_face_options.py ├── face_options.py ├── material_options.py ├── medical_material_options.py └── medical_options.py ├── scripts └── train_test.sh ├── test.py ├── train.py └── util ├── data_processor.py ├── evaluation.py ├── image_pool.py ├── image_tools.py ├── metrics.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/README.md -------------------------------------------------------------------------------- /algorithm/fedavg/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/fedavg/client.py -------------------------------------------------------------------------------- /algorithm/fedavg/fedavg_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/fedavg/fedavg_trainer.py -------------------------------------------------------------------------------- /algorithm/fedavg/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/fedavg/optim.py -------------------------------------------------------------------------------- /algorithm/fedavg/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/fedavg/util.py -------------------------------------------------------------------------------- /algorithm/feddc/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/feddc/client.py -------------------------------------------------------------------------------- /algorithm/feddc/feddc_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/feddc/feddc_trainer.py -------------------------------------------------------------------------------- /algorithm/feddc/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/feddc/util.py -------------------------------------------------------------------------------- /algorithm/feddyn/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/feddyn/client.py -------------------------------------------------------------------------------- /algorithm/feddyn/comm_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/feddyn/comm_helpers.py -------------------------------------------------------------------------------- /algorithm/feddyn/feddyn_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/feddyn/feddyn_trainer.py -------------------------------------------------------------------------------- /algorithm/feddyn/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/feddyn/optim.py -------------------------------------------------------------------------------- /algorithm/fedprox/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/fedprox/client.py -------------------------------------------------------------------------------- /algorithm/fedprox/comm_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/fedprox/comm_helpers.py -------------------------------------------------------------------------------- /algorithm/fedprox/fedprox_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/fedprox/fedprox_trainer.py -------------------------------------------------------------------------------- /algorithm/fedprox/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/fedprox/optim.py -------------------------------------------------------------------------------- /algorithm/local/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/local/client.py -------------------------------------------------------------------------------- /algorithm/local/local_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/local/local_trainer.py -------------------------------------------------------------------------------- /algorithm/local/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/local/optim.py -------------------------------------------------------------------------------- /algorithm/local/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/algorithm/local/util.py -------------------------------------------------------------------------------- /code_structure.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/code_structure.md -------------------------------------------------------------------------------- /core/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/core/base_dataset.py -------------------------------------------------------------------------------- /core/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/core/base_model.py -------------------------------------------------------------------------------- /core/base_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/core/base_network.py -------------------------------------------------------------------------------- /core/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/core/logger.py -------------------------------------------------------------------------------- /core/praser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/core/praser.py -------------------------------------------------------------------------------- /core/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/core/util.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/data_processing/data_partion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/data/data_processing/data_partion.py -------------------------------------------------------------------------------- /data/dataloader/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/data/dataloader/data_loader.py -------------------------------------------------------------------------------- /data/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/data/dataset/__init__.py -------------------------------------------------------------------------------- /data/dataset/aligned_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/data/dataset/aligned_dataset.py -------------------------------------------------------------------------------- /data/dataset/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/data/dataset/base_dataset.py -------------------------------------------------------------------------------- /data/dataset/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/data/dataset/image_folder.py -------------------------------------------------------------------------------- /data/dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/data/dataset/utils.py -------------------------------------------------------------------------------- /ddpm/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/LICENSE -------------------------------------------------------------------------------- /ddpm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/README.md -------------------------------------------------------------------------------- /ddpm/README.md.backup: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/README.md.backup -------------------------------------------------------------------------------- /ddpm/README_cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/README_cn.md -------------------------------------------------------------------------------- /ddpm/config/AAF_config.7z: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/AAF_config.7z -------------------------------------------------------------------------------- /ddpm/config/AAFnew0_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/AAFnew0_train.json -------------------------------------------------------------------------------- /ddpm/config/CelebaHQ_face_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/CelebaHQ_face_train.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_face0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_face0.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_face1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_face1.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_face2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_face2.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_material0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_material0.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_material0_seperate.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_material0_seperate.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_material0_seperate_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_material0_seperate_test.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_material1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_material1.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_material1_seperate.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_material1_seperate.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_material1_seperate_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_material1_seperate_test.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_medical0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_medical0.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_medical0_material1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_medical0_material1.json -------------------------------------------------------------------------------- /ddpm/config/labeltoimage_medical1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/config/labeltoimage_medical1.json -------------------------------------------------------------------------------- /ddpm/core/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/core/base_dataset.py -------------------------------------------------------------------------------- /ddpm/core/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/core/base_model.py -------------------------------------------------------------------------------- /ddpm/core/base_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/core/base_network.py -------------------------------------------------------------------------------- /ddpm/core/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/core/logger.py -------------------------------------------------------------------------------- /ddpm/core/praser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/core/praser.py -------------------------------------------------------------------------------- /ddpm/core/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/core/util.py -------------------------------------------------------------------------------- /ddpm/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/data/__init__.py -------------------------------------------------------------------------------- /ddpm/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/data/dataset.py -------------------------------------------------------------------------------- /ddpm/data/util/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/data/util/auto_augment.py -------------------------------------------------------------------------------- /ddpm/data/util/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/data/util/mask.py -------------------------------------------------------------------------------- /ddpm/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/eval.py -------------------------------------------------------------------------------- /ddpm/misc/Palette Image-to-Image Diffusion Models.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/Palette Image-to-Image Diffusion Models.pdf -------------------------------------------------------------------------------- /ddpm/misc/image/Mask_Places365_test_00143399.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Mask_Places365_test_00143399.jpg -------------------------------------------------------------------------------- /ddpm/misc/image/Mask_Places365_test_00144085.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Mask_Places365_test_00144085.jpg -------------------------------------------------------------------------------- /ddpm/misc/image/Mask_Places365_test_00209019.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Mask_Places365_test_00209019.jpg -------------------------------------------------------------------------------- /ddpm/misc/image/Mask_Places365_test_00263905.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Mask_Places365_test_00263905.jpg -------------------------------------------------------------------------------- /ddpm/misc/image/Out_Places365_test_00143399.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Out_Places365_test_00143399.jpg -------------------------------------------------------------------------------- /ddpm/misc/image/Out_Places365_test_00144085.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Out_Places365_test_00144085.jpg -------------------------------------------------------------------------------- /ddpm/misc/image/Out_Places365_test_00209019.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Out_Places365_test_00209019.jpg -------------------------------------------------------------------------------- /ddpm/misc/image/Out_Places365_test_00263905.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Out_Places365_test_00263905.jpg -------------------------------------------------------------------------------- /ddpm/misc/image/Process_02323.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Process_02323.jpg -------------------------------------------------------------------------------- /ddpm/misc/image/Process_26190.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Process_26190.jpg -------------------------------------------------------------------------------- /ddpm/misc/image/Process_Places365_test_00042384.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Process_Places365_test_00042384.jpg -------------------------------------------------------------------------------- /ddpm/misc/image/Process_Places365_test_00309553.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/misc/image/Process_Places365_test_00309553.jpg -------------------------------------------------------------------------------- /ddpm/models.join/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models.join/__init__.py -------------------------------------------------------------------------------- /ddpm/models.join/guided_diffusion_modules/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models.join/guided_diffusion_modules/nn.py -------------------------------------------------------------------------------- /ddpm/models.join/guided_diffusion_modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models.join/guided_diffusion_modules/unet.py -------------------------------------------------------------------------------- /ddpm/models.join/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models.join/loss.py -------------------------------------------------------------------------------- /ddpm/models.join/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models.join/metric.py -------------------------------------------------------------------------------- /ddpm/models.join/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models.join/model.py -------------------------------------------------------------------------------- /ddpm/models.join/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models.join/network.py -------------------------------------------------------------------------------- /ddpm/models.join/sr3_modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models.join/sr3_modules/unet.py -------------------------------------------------------------------------------- /ddpm/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models/__init__.py -------------------------------------------------------------------------------- /ddpm/models/cnn_modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ddpm/models/cnn_modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models/cnn_modules/unet.py -------------------------------------------------------------------------------- /ddpm/models/guided_diffusion_modules/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models/guided_diffusion_modules/nn.py -------------------------------------------------------------------------------- /ddpm/models/guided_diffusion_modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models/guided_diffusion_modules/unet.py -------------------------------------------------------------------------------- /ddpm/models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models/loss.py -------------------------------------------------------------------------------- /ddpm/models/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models/metric.py -------------------------------------------------------------------------------- /ddpm/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models/model.py -------------------------------------------------------------------------------- /ddpm/models/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models/network.py -------------------------------------------------------------------------------- /ddpm/models/sr3_modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/models/sr3_modules/unet.py -------------------------------------------------------------------------------- /ddpm/preprocess/mirflickr25k_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/preprocess/mirflickr25k_preprocess.py -------------------------------------------------------------------------------- /ddpm/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/requirements.txt -------------------------------------------------------------------------------- /ddpm/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/run.py -------------------------------------------------------------------------------- /ddpm/scripts/run_AAF.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/scripts/run_AAF.sh -------------------------------------------------------------------------------- /ddpm/slurm/inpainting_places2.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/slurm/inpainting_places2.slurm -------------------------------------------------------------------------------- /ddpm/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/test.sh -------------------------------------------------------------------------------- /ddpm/tools/change_image_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/change_image_format.py -------------------------------------------------------------------------------- /ddpm/tools/copy_fake_image_to_target_dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/copy_fake_image_to_target_dir.py -------------------------------------------------------------------------------- /ddpm/tools/copy_fake_image_to_target_dir_join.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/copy_fake_image_to_target_dir_join.py -------------------------------------------------------------------------------- /ddpm/tools/create_config_for_diffusion_seperate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/create_config_for_diffusion_seperate.py -------------------------------------------------------------------------------- /ddpm/tools/create_flist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/create_flist.py -------------------------------------------------------------------------------- /ddpm/tools/create_test_config_for_diffusion_join.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/create_test_config_for_diffusion_join.py -------------------------------------------------------------------------------- /ddpm/tools/create_test_config_for_diffusion_seperate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/create_test_config_for_diffusion_seperate.py -------------------------------------------------------------------------------- /ddpm/tools/data_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/data_counter.py -------------------------------------------------------------------------------- /ddpm/tools/data_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/data_merge.py -------------------------------------------------------------------------------- /ddpm/tools/data_partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/data_partition.py -------------------------------------------------------------------------------- /ddpm/tools/data_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/data_select.py -------------------------------------------------------------------------------- /ddpm/tools/delete_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/delete_file.py -------------------------------------------------------------------------------- /ddpm/tools/delete_pkl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/delete_pkl.py -------------------------------------------------------------------------------- /ddpm/tools/draw_face_color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/draw_face_color.py -------------------------------------------------------------------------------- /ddpm/tools/get_pred_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/get_pred_evaluation.py -------------------------------------------------------------------------------- /ddpm/tools/get_style_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/get_style_images.py -------------------------------------------------------------------------------- /ddpm/tools/image_collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/image_collection.py -------------------------------------------------------------------------------- /ddpm/tools/mask_id_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/mask_id_map.py -------------------------------------------------------------------------------- /ddpm/tools/multi_dir_to_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/multi_dir_to_single.py -------------------------------------------------------------------------------- /ddpm/tools/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/predict.py -------------------------------------------------------------------------------- /ddpm/tools/predict_face_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/predict_face_seg.py -------------------------------------------------------------------------------- /ddpm/tools/t-SNE_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/tools/t-SNE_test.py -------------------------------------------------------------------------------- /ddpm/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/ddpm/train.sh -------------------------------------------------------------------------------- /docs/Abstract.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/docs/Abstract.png -------------------------------------------------------------------------------- /docs/Network.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/docs/Network.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/base_model.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/AAFnew0_join_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/AAFnew0_join_train.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/colorization_mirflickr25k.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/colorization_mirflickr25k.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/inpainting_celebahq.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/inpainting_celebahq.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/inpainting_places2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/inpainting_places2.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/labeltoimage.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/labeltoimage.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/labeltoimage_aaf_face_new.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/labeltoimage_aaf_face_new.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/labeltoimage_face0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/labeltoimage_face0.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/labeltoimage_face1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/labeltoimage_face1.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/labeltoimage_face2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/labeltoimage_face2.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/labeltoimage_material0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/labeltoimage_material0.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/labeltoimage_material1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/labeltoimage_material1.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/labeltoimage_medical0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/labeltoimage_medical0.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/labeltoimage_medical1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/labeltoimage_medical1.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/config/uncropping_places2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/config/uncropping_places2.json -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/core/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/core/base_dataset.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/core/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/core/base_model.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/core/base_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/core/base_network.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/core/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/core/logger.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/core/praser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/core/praser.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/core/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/core/util.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/data/__init__.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/data/dataset.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/data/util/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/data/util/auto_augment.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/data/util/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/data/util/mask.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/ddpm_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/ddpm_config.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/models/__init__.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/models/guided_diffusion_modules/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/models/guided_diffusion_modules/nn.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/models/guided_diffusion_modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/models/guided_diffusion_modules/unet.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/models/loss.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/models/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/models/metric.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/models/model.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/models/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/models/network.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/models/sr3_modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/models/sr3_modules/unet.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/ddpm_models/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/ddpm_models/network.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/fedst_ddpm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/fedst_ddpm_model.py -------------------------------------------------------------------------------- /models/fedst_ddpm_model/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/fedst_ddpm_model/networks.py -------------------------------------------------------------------------------- /models/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/losses/losses.py -------------------------------------------------------------------------------- /models/unet_model/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/unet_model/base_model.py -------------------------------------------------------------------------------- /models/unet_model/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/unet_model/networks.py -------------------------------------------------------------------------------- /models/unet_model/unet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/models/unet_model/unet_model.py -------------------------------------------------------------------------------- /opt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /opt/aaf_face_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/opt/aaf_face_options.py -------------------------------------------------------------------------------- /opt/face_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/opt/face_options.py -------------------------------------------------------------------------------- /opt/material_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/opt/material_options.py -------------------------------------------------------------------------------- /opt/medical_material_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/opt/medical_material_options.py -------------------------------------------------------------------------------- /opt/medical_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/opt/medical_options.py -------------------------------------------------------------------------------- /scripts/train_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/scripts/train_test.sh -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/train.py -------------------------------------------------------------------------------- /util/data_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/util/data_processor.py -------------------------------------------------------------------------------- /util/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/util/evaluation.py -------------------------------------------------------------------------------- /util/image_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/util/image_pool.py -------------------------------------------------------------------------------- /util/image_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/util/image_tools.py -------------------------------------------------------------------------------- /util/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/util/metrics.py -------------------------------------------------------------------------------- /util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YoferChen/FedST/HEAD/util/util.py --------------------------------------------------------------------------------