Init commit

.gitignore

@@ -0,0 +1,9 @@
+.idea/*
+*__pycache__*
+weights
+experiments
+pretrained_models
+*.so
+*.ipynb_checkpoints*
+*.yml
+*.json

Dockerfile

@@ -0,0 +1,37 @@
+FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+        build-essential \
+	    git \
+	    curl \
+        libglib2.0-0 \
+        software-properties-common \
+        python3.6-dev \
+        python3-pip \
+        python3-tk
+
+WORKDIR /tmp
+
+RUN pip3 install --upgrade pip
+RUN pip3 install setuptools
+RUN pip3 install matplotlib numpy pandas scipy tqdm pyyaml easydict scikit-image bridson Pillow ninja
+RUN pip3 install imgaug mxboard graphviz
+RUN pip3 install albumentations --no-deps
+RUN pip3 install opencv-python-headless
+RUN pip3 install Cython
+RUN pip3 install torch
+RUN pip3 install torchvision
+RUN pip3 install scikit-learn
+RUN pip3 install tensorboard
+
+RUN mkdir /work
+WORKDIR /work
+RUN chmod -R 777 /work && chmod -R 777 /root
+
+ENV TINI_VERSION v0.18.0
+ADD https://github.com/krallin/tini/releases/download/${TINI_VERSION}/tini /usr/bin/tini
+RUN chmod +x /usr/bin/tini
+ENTRYPOINT [ "/usr/bin/tini", "--" ]
+CMD [ "/bin/bash" ]

LICENSE

@@ -0,0 +1,21 @@
+The MIT License
+
+Copyright (c) 2021 Samsung Electronics Co., Ltd.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

demo.py

@@ -0,0 +1,57 @@
+import argparse
+import tkinter as tk
+
+import torch
+
+from isegm.utils import exp
+from isegm.inference import utils
+from interactive_demo.app import InteractiveDemoApp
+
+
+def main():
+    args, cfg = parse_args()
+
+    torch.backends.cudnn.deterministic = True
+    checkpoint_path = utils.find_checkpoint(cfg.INTERACTIVE_MODELS_PATH, args.checkpoint)
+    model = utils.load_is_model(checkpoint_path, args.device, cpu_dist_maps=True)
+
+    root = tk.Tk()
+    root.minsize(960, 480)
+    app = InteractiveDemoApp(root, args, model)
+    root.deiconify()
+    app.mainloop()
+
+
+def parse_args():
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('--checkpoint', type=str, required=True,
+                        help='The path to the checkpoint. '
+                             'This can be a relative path (relative to cfg.INTERACTIVE_MODELS_PATH) '
+                             'or an absolute path. The file extension can be omitted.')
+
+    parser.add_argument('--gpu', type=int, default=0,
+                        help='Id of GPU to use.')
+
+    parser.add_argument('--cpu', action='store_true', default=False,
+                        help='Use only CPU for inference.')
+
+    parser.add_argument('--limit-longest-size', type=int, default=800,
+                        help='If the largest side of an image exceeds this value, '
+                             'it is resized so that its largest side is equal to this value.')
+
+    parser.add_argument('--cfg', type=str, default="config.yml",
+                        help='The path to the config file.')
+
+    args = parser.parse_args()
+    if args.cpu:
+        args.device =torch.device('cpu')
+    else:
+        args.device = torch.device(f'cuda:{args.gpu}')
+    cfg = exp.load_config_file(args.cfg, return_edict=True)
+
+    return args, cfg
+
+
+if __name__ == '__main__':
+    main()