add large model

README.md

@@ -1,6 +1,6 @@
 ---
 title: Image Deblurring
-emoji: 💻
+emoji: 📷
 colorFrom: gray
 colorTo: pink
 sdk: gradio

app.py

@@ -1,17 +1,17 @@
 import sys
 sys.path.append("src")
-from interactive_pipe import interactive_pipeline
-from rstor.analyzis.interactive.pipelines import natural_inference_pipeline, morph_canvas, CANVAS
-from rstor.analyzis.interactive.model_selection import get_default_models
-from pathlib import Path
-from rstor.analyzis.parser import get_parser
-import argparse
-from batch_processing import Batch
-from interactive_pipe.data_objects.image import Image
-from rstor.analyzis.interactive.images import image_selector
-from rstor.analyzis.interactive.crop import plug_crop_selector
-from rstor.analyzis.interactive.metrics import plug_configure_metrics
 from interactive_pipe import interactive, KeyboardControl
+from rstor.analyzis.interactive.metrics import plug_configure_metrics
+from rstor.analyzis.interactive.crop import plug_crop_selector
+from rstor.analyzis.interactive.images import image_selector
+from interactive_pipe.data_objects.image import Image
+from batch_processing import Batch
+import argparse
+from rstor.analyzis.parser import get_parser
+from pathlib import Path
+from rstor.analyzis.interactive.model_selection import get_default_models
+from rstor.analyzis.interactive.pipelines import natural_inference_pipeline, morph_canvas, CANVAS
+from interactive_pipe import interactive_pipeline
 
 
 def plug_morph_canvas():
@@ -48,8 +48,8 @@ def main(argv):
         image_control = (0, [0, len(img_list)-1])
     interactive(image_index=image_control)(image_selector)
     plug_crop_selector(num_pad=args.keyboard)
-    plug_configure_metrics(key_shortcut="a")  # "a" if args.keyboard else None)
-    plug_morph_canvas()
+    # plug_configure_metrics(key_shortcut="a")  # "a" if args.keyboard else None)
+    # plug_morph_canvas()
     model_dict = get_default_models(args.experiments, Path(args.models_storage), keyboard_control=args.keyboard)
     interactive_pipeline(
         gui=args.backend,
@@ -63,4 +63,4 @@ def main(argv):
 
 if __name__ == "__main__":
     # main(sys.argv[1:])
-    main(["-e", "6002", "-i", "__dataset/sample/*.*g",  "-b","gradio"])
+    main(["-e", "6002", "5002", "-i", "__dataset/sample/*.*g",  "-b", "gradio"])

scripts/configuration.py

@@ -14,5 +14,5 @@ WANDBSPACE = "deblur-from-deadleaves"
 TRAIN_SCRIPT = "scripts/train.py"  # Location of the training script
 
 ROOT_DIR = Path(__file__).parent
-OUTPUT_FOLDER_NAME = "__output"
+OUTPUT_FOLDER_NAME = "output"
 INFERENCE_FOLDER_NAME = "__inference"

scripts/output/5002/best_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:510eaf7c70702efc53afca96a5edf2ee83127210d8e8914e5e6c44bfc1aeb83b
+size 165984598

src/rstor/analyzis/interactive/model_selection.py

@@ -8,7 +8,7 @@ from pathlib import Path
 from typing import List, Tuple
 
 from interactive_pipe import interactive
-MODELS_PATH = Path("scripts")/"__output"
+MODELS_PATH = Path("scripts")/"output"
 
 
 def model_selector(models_dict: dict, global_params={}, model_name="vanilla"):