Initial commit with some structure

cdim/noise_configs/gaussian_noise.yaml

@@ -0,0 +1,2 @@
+name: gaussian
+sigma: 0.05

cdim/noise_configs/poisson_noise_config.yaml

@@ -0,0 +1,2 @@
+name: poisson
+rate: 0.1

environment.yml

@@ -0,0 +1,6 @@
+name: cdim
+channels:
+  - defaults
+dependencies:
+  - python=3.11.5
+prefix: /gscratch/realitylab/vjayaram/miniconda3/envs/cdim

inference.py

@@ -0,0 +1,20 @@
+import argparse
+import os
+
+
+def main(args):
+    os.makedirs(args.output_dir, exist_ok=True)
+
+    
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument("input_image", type=str)
+    parser.add_argument("T", type=int)
+    parser.add_argument("K", type=int)
+    parser.add_argument("model", type=str)
+    parser.add_argument("operator_config", type=str)
+    parser.add_argument("noise_config", type=str)
+    parser.add_argument("--output-dir", default=".", type=str)
+    main(parser.parse_args())