├── .gitignore ├── .project-root ├── README.md ├── assets ├── comparison.png ├── num_parameters.png └── visual_comparison.png ├── checkpoints └── .gitkeep ├── configs ├── __init__.py ├── callbacks │ ├── default.yaml │ ├── early_stopping.yaml │ ├── model_checkpoint.yaml │ ├── model_summary.yaml │ └── rich_progress_bar.yaml ├── data │ ├── default.yaml │ └── image.yaml ├── eval.yaml ├── experiment │ └── wz │ │ ├── wz_eval.yaml │ │ └── wz_train_bw.yaml ├── extras │ └── default.yaml ├── hydra │ └── default.yaml ├── logger │ ├── comet.yaml │ └── mlflow.yaml ├── model │ ├── base_deepjscc.yaml │ ├── channel │ │ ├── awgn.yaml │ │ └── perfect.yaml │ └── loss │ │ ├── lpips.yaml │ │ ├── mse.yaml │ │ ├── mse_lpips.yaml │ │ ├── msssim.yaml │ │ └── ssim.yaml ├── paths │ └── default.yaml ├── train.yaml └── trainer │ ├── cpu.yaml │ ├── ddp.yaml │ ├── ddp_sim.yaml │ ├── default.yaml │ ├── gpu.yaml │ └── mps.yaml ├── data ├── .gitkeep ├── KITTI_stereo_test.txt ├── KITTI_stereo_train.txt └── KITTI_stereo_val.txt ├── requirements.txt ├── results ├── Cityscape_16_DeepJSCCWZ.csv ├── Cityscape_16_DeepJSCCWZBaseline.csv ├── Cityscape_16_DeepJSCCWZBaseline2.csv ├── Cityscape_16_DeepJSCCWZJoint2.csv ├── Cityscape_32_DeepJSCCWZ.csv ├── Cityscape_32_DeepJSCCWZBaseline.csv ├── Cityscape_32_DeepJSCCWZBaseline2.csv ├── Cityscape_32_DeepJSCCWZJoint2.csv ├── KittiStereo_16_DeepJSCCWZ.csv ├── KittiStereo_16_DeepJSCCWZBaseline.csv ├── KittiStereo_16_DeepJSCCWZBaseline2.csv ├── KittiStereo_16_DeepJSCCWZJoint2.csv ├── KittiStereo_32_DeepJSCCWZ.csv ├── KittiStereo_32_DeepJSCCWZBaseline.csv ├── KittiStereo_32_DeepJSCCWZBaseline2.csv └── KittiStereo_32_DeepJSCCWZJoint2.csv └── src ├── __init__.py ├── data ├── __init__.py ├── base.py ├── image.py └── stereo.py ├── eval.py ├── models ├── __init__.py ├── base_deepjscc.py ├── components │ ├── __init__.py │ ├── afmodule.py │ ├── channels.py │ ├── compressai.py │ ├── csi_generators.py │ ├── losses.py │ ├── metrics.py │ └── power_constraints.py ├── deepjscc.py ├── nets │ ├── __init__.py │ └── utils.py └── wz │ ├── __init__.py │ ├── deepjscc_wz.py │ ├── deepjscc_wz_baseline.py │ ├── deepjscc_wz_baseline2.py │ └── deepjscc_wz_joint2.py ├── train.py └── utils ├── __init__.py ├── img_utils.py ├── pylogger.py ├── rich_utils.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/.gitignore -------------------------------------------------------------------------------- /.project-root: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/.project-root -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/README.md -------------------------------------------------------------------------------- /assets/comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/assets/comparison.png -------------------------------------------------------------------------------- /assets/num_parameters.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/assets/num_parameters.png -------------------------------------------------------------------------------- /assets/visual_comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/assets/visual_comparison.png -------------------------------------------------------------------------------- /checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/__init__.py -------------------------------------------------------------------------------- /configs/callbacks/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/callbacks/default.yaml -------------------------------------------------------------------------------- /configs/callbacks/early_stopping.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/callbacks/early_stopping.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_checkpoint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/callbacks/model_checkpoint.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_summary.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/callbacks/model_summary.yaml -------------------------------------------------------------------------------- /configs/callbacks/rich_progress_bar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/callbacks/rich_progress_bar.yaml -------------------------------------------------------------------------------- /configs/data/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/data/default.yaml -------------------------------------------------------------------------------- /configs/data/image.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/data/image.yaml -------------------------------------------------------------------------------- /configs/eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/eval.yaml -------------------------------------------------------------------------------- /configs/experiment/wz/wz_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/experiment/wz/wz_eval.yaml -------------------------------------------------------------------------------- /configs/experiment/wz/wz_train_bw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/experiment/wz/wz_train_bw.yaml -------------------------------------------------------------------------------- /configs/extras/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/extras/default.yaml -------------------------------------------------------------------------------- /configs/hydra/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/hydra/default.yaml -------------------------------------------------------------------------------- /configs/logger/comet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/logger/comet.yaml -------------------------------------------------------------------------------- /configs/logger/mlflow.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/logger/mlflow.yaml -------------------------------------------------------------------------------- /configs/model/base_deepjscc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/model/base_deepjscc.yaml -------------------------------------------------------------------------------- /configs/model/channel/awgn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/model/channel/awgn.yaml -------------------------------------------------------------------------------- /configs/model/channel/perfect.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/model/channel/perfect.yaml -------------------------------------------------------------------------------- /configs/model/loss/lpips.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/model/loss/lpips.yaml -------------------------------------------------------------------------------- /configs/model/loss/mse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/model/loss/mse.yaml -------------------------------------------------------------------------------- /configs/model/loss/mse_lpips.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/model/loss/mse_lpips.yaml -------------------------------------------------------------------------------- /configs/model/loss/msssim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/model/loss/msssim.yaml -------------------------------------------------------------------------------- /configs/model/loss/ssim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/model/loss/ssim.yaml -------------------------------------------------------------------------------- /configs/paths/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/paths/default.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /configs/trainer/cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/trainer/cpu.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/trainer/ddp.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp_sim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/trainer/ddp_sim.yaml -------------------------------------------------------------------------------- /configs/trainer/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/trainer/default.yaml -------------------------------------------------------------------------------- /configs/trainer/gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/trainer/gpu.yaml -------------------------------------------------------------------------------- /configs/trainer/mps.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/configs/trainer/mps.yaml -------------------------------------------------------------------------------- /data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/KITTI_stereo_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/data/KITTI_stereo_test.txt -------------------------------------------------------------------------------- /data/KITTI_stereo_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/data/KITTI_stereo_train.txt -------------------------------------------------------------------------------- /data/KITTI_stereo_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/data/KITTI_stereo_val.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/Cityscape_16_DeepJSCCWZ.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/Cityscape_16_DeepJSCCWZ.csv -------------------------------------------------------------------------------- /results/Cityscape_16_DeepJSCCWZBaseline.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/Cityscape_16_DeepJSCCWZBaseline.csv -------------------------------------------------------------------------------- /results/Cityscape_16_DeepJSCCWZBaseline2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/Cityscape_16_DeepJSCCWZBaseline2.csv -------------------------------------------------------------------------------- /results/Cityscape_16_DeepJSCCWZJoint2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/Cityscape_16_DeepJSCCWZJoint2.csv -------------------------------------------------------------------------------- /results/Cityscape_32_DeepJSCCWZ.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/Cityscape_32_DeepJSCCWZ.csv -------------------------------------------------------------------------------- /results/Cityscape_32_DeepJSCCWZBaseline.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/Cityscape_32_DeepJSCCWZBaseline.csv -------------------------------------------------------------------------------- /results/Cityscape_32_DeepJSCCWZBaseline2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/Cityscape_32_DeepJSCCWZBaseline2.csv -------------------------------------------------------------------------------- /results/Cityscape_32_DeepJSCCWZJoint2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/Cityscape_32_DeepJSCCWZJoint2.csv -------------------------------------------------------------------------------- /results/KittiStereo_16_DeepJSCCWZ.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/KittiStereo_16_DeepJSCCWZ.csv -------------------------------------------------------------------------------- /results/KittiStereo_16_DeepJSCCWZBaseline.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/KittiStereo_16_DeepJSCCWZBaseline.csv -------------------------------------------------------------------------------- /results/KittiStereo_16_DeepJSCCWZBaseline2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/KittiStereo_16_DeepJSCCWZBaseline2.csv -------------------------------------------------------------------------------- /results/KittiStereo_16_DeepJSCCWZJoint2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/KittiStereo_16_DeepJSCCWZJoint2.csv -------------------------------------------------------------------------------- /results/KittiStereo_32_DeepJSCCWZ.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/KittiStereo_32_DeepJSCCWZ.csv -------------------------------------------------------------------------------- /results/KittiStereo_32_DeepJSCCWZBaseline.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/KittiStereo_32_DeepJSCCWZBaseline.csv -------------------------------------------------------------------------------- /results/KittiStereo_32_DeepJSCCWZBaseline2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/KittiStereo_32_DeepJSCCWZBaseline2.csv -------------------------------------------------------------------------------- /results/KittiStereo_32_DeepJSCCWZJoint2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/results/KittiStereo_32_DeepJSCCWZJoint2.csv -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/data/base.py -------------------------------------------------------------------------------- /src/data/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/data/image.py -------------------------------------------------------------------------------- /src/data/stereo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/data/stereo.py -------------------------------------------------------------------------------- /src/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/eval.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/base_deepjscc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/base_deepjscc.py -------------------------------------------------------------------------------- /src/models/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/afmodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/components/afmodule.py -------------------------------------------------------------------------------- /src/models/components/channels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/components/channels.py -------------------------------------------------------------------------------- /src/models/components/compressai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/components/compressai.py -------------------------------------------------------------------------------- /src/models/components/csi_generators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/components/csi_generators.py -------------------------------------------------------------------------------- /src/models/components/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/components/losses.py -------------------------------------------------------------------------------- /src/models/components/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/components/metrics.py -------------------------------------------------------------------------------- /src/models/components/power_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/components/power_constraints.py -------------------------------------------------------------------------------- /src/models/deepjscc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/deepjscc.py -------------------------------------------------------------------------------- /src/models/nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/nets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/nets/utils.py -------------------------------------------------------------------------------- /src/models/wz/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/wz/deepjscc_wz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/wz/deepjscc_wz.py -------------------------------------------------------------------------------- /src/models/wz/deepjscc_wz_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/wz/deepjscc_wz_baseline.py -------------------------------------------------------------------------------- /src/models/wz/deepjscc_wz_baseline2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/wz/deepjscc_wz_baseline2.py -------------------------------------------------------------------------------- /src/models/wz/deepjscc_wz_joint2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/models/wz/deepjscc_wz_joint2.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/img_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/utils/img_utils.py -------------------------------------------------------------------------------- /src/utils/pylogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/utils/pylogger.py -------------------------------------------------------------------------------- /src/utils/rich_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/utils/rich_utils.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ipc-lab/deepjscc-wz/HEAD/src/utils/utils.py --------------------------------------------------------------------------------