├── LICENSE ├── README.md ├── __pycache__ ├── Utils_2.cpython-38.pyc ├── dataset.cpython-38.pyc └── utils.cpython-38.pyc ├── app.py ├── dataset.py ├── demo.py ├── demo ├── results │ ├── ours_giraffe_grass_zebra.png │ ├── ours_kite_grass_woman.png │ ├── ours_kite_tree_person.png │ └── threshold │ │ ├── bench_th_5.png │ │ ├── bench_th_6.png │ │ ├── bench_th_7.png │ │ ├── bench_th_8.png │ │ ├── bench_th_9.png │ │ ├── tree_th_5.png │ │ ├── tree_th_6.png │ │ ├── tree_th_7.png │ │ ├── tree_th_8.png │ │ └── tree_th_9.png ├── sketch_1.png ├── sketch_2.png ├── sketch_3.png └── threshold_example.png ├── figs ├── inference.gif ├── teaser.png └── thresholding.png ├── models ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── __init__.cpython-39.pyc │ ├── auxilary.cpython-38.pyc │ ├── build_model.cpython-38.pyc │ ├── ca.cpython-38.pyc │ ├── clip.cpython-38.pyc │ ├── clip.cpython-39.pyc │ ├── clip_model.cpython-38.pyc │ ├── clip_surgery_model.cpython-38.pyc │ ├── our_model.cpython-38.pyc │ └── simple_tokenizer.cpython-38.pyc ├── auxilary.py ├── bpe_simple_vocab_16e6.txt.gz ├── build_model.py ├── ca.py ├── clip.py ├── clip_model.py ├── our_model.py └── simple_tokenizer.py ├── output.png ├── test.py ├── train.py ├── utils.py └── vpt ├── __pycache__ └── launch.cpython-38.pyc ├── configs ├── base-prompt.yaml └── prompt │ └── cub.yaml ├── launch.py └── src ├── configs ├── __pycache__ │ ├── config.cpython-38.pyc │ └── config_node.cpython-38.pyc ├── config.py ├── config_node.py └── vit_configs.py └── utils ├── __pycache__ ├── distributed.cpython-38.pyc ├── file_io.cpython-38.pyc └── logging.cpython-38.pyc ├── distributed.py ├── file_io.py └── logging.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/README.md -------------------------------------------------------------------------------- /__pycache__/Utils_2.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/__pycache__/Utils_2.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/dataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/__pycache__/dataset.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/app.py -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/dataset.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo.py -------------------------------------------------------------------------------- /demo/results/ours_giraffe_grass_zebra.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/ours_giraffe_grass_zebra.png -------------------------------------------------------------------------------- /demo/results/ours_kite_grass_woman.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/ours_kite_grass_woman.png -------------------------------------------------------------------------------- /demo/results/ours_kite_tree_person.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/ours_kite_tree_person.png -------------------------------------------------------------------------------- /demo/results/threshold/bench_th_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/threshold/bench_th_5.png -------------------------------------------------------------------------------- /demo/results/threshold/bench_th_6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/threshold/bench_th_6.png -------------------------------------------------------------------------------- /demo/results/threshold/bench_th_7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/threshold/bench_th_7.png -------------------------------------------------------------------------------- /demo/results/threshold/bench_th_8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/threshold/bench_th_8.png -------------------------------------------------------------------------------- /demo/results/threshold/bench_th_9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/threshold/bench_th_9.png -------------------------------------------------------------------------------- /demo/results/threshold/tree_th_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/threshold/tree_th_5.png -------------------------------------------------------------------------------- /demo/results/threshold/tree_th_6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/threshold/tree_th_6.png -------------------------------------------------------------------------------- /demo/results/threshold/tree_th_7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/threshold/tree_th_7.png -------------------------------------------------------------------------------- /demo/results/threshold/tree_th_8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/threshold/tree_th_8.png -------------------------------------------------------------------------------- /demo/results/threshold/tree_th_9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/results/threshold/tree_th_9.png -------------------------------------------------------------------------------- /demo/sketch_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/sketch_1.png -------------------------------------------------------------------------------- /demo/sketch_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/sketch_2.png -------------------------------------------------------------------------------- /demo/sketch_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/sketch_3.png -------------------------------------------------------------------------------- /demo/threshold_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/demo/threshold_example.png -------------------------------------------------------------------------------- /figs/inference.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/figs/inference.gif -------------------------------------------------------------------------------- /figs/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/figs/teaser.png -------------------------------------------------------------------------------- /figs/thresholding.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/figs/thresholding.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/auxilary.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/__pycache__/auxilary.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/build_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/__pycache__/build_model.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/ca.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/__pycache__/ca.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/clip.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/__pycache__/clip.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/clip.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/__pycache__/clip.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/clip_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/__pycache__/clip_model.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/clip_surgery_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/__pycache__/clip_surgery_model.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/our_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/__pycache__/our_model.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/simple_tokenizer.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/__pycache__/simple_tokenizer.cpython-38.pyc -------------------------------------------------------------------------------- /models/auxilary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/auxilary.py -------------------------------------------------------------------------------- /models/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /models/build_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/build_model.py -------------------------------------------------------------------------------- /models/ca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/ca.py -------------------------------------------------------------------------------- /models/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/clip.py -------------------------------------------------------------------------------- /models/clip_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/clip_model.py -------------------------------------------------------------------------------- /models/our_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/our_model.py -------------------------------------------------------------------------------- /models/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/models/simple_tokenizer.py -------------------------------------------------------------------------------- /output.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/output.png -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/utils.py -------------------------------------------------------------------------------- /vpt/__pycache__/launch.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/__pycache__/launch.cpython-38.pyc -------------------------------------------------------------------------------- /vpt/configs/base-prompt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/configs/base-prompt.yaml -------------------------------------------------------------------------------- /vpt/configs/prompt/cub.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/configs/prompt/cub.yaml -------------------------------------------------------------------------------- /vpt/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/launch.py -------------------------------------------------------------------------------- /vpt/src/configs/__pycache__/config.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/src/configs/__pycache__/config.cpython-38.pyc -------------------------------------------------------------------------------- /vpt/src/configs/__pycache__/config_node.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/src/configs/__pycache__/config_node.cpython-38.pyc -------------------------------------------------------------------------------- /vpt/src/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/src/configs/config.py -------------------------------------------------------------------------------- /vpt/src/configs/config_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/src/configs/config_node.py -------------------------------------------------------------------------------- /vpt/src/configs/vit_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/src/configs/vit_configs.py -------------------------------------------------------------------------------- /vpt/src/utils/__pycache__/distributed.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/src/utils/__pycache__/distributed.cpython-38.pyc -------------------------------------------------------------------------------- /vpt/src/utils/__pycache__/file_io.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/src/utils/__pycache__/file_io.cpython-38.pyc -------------------------------------------------------------------------------- /vpt/src/utils/__pycache__/logging.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/src/utils/__pycache__/logging.cpython-38.pyc -------------------------------------------------------------------------------- /vpt/src/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/src/utils/distributed.py -------------------------------------------------------------------------------- /vpt/src/utils/file_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/src/utils/file_io.py -------------------------------------------------------------------------------- /vpt/src/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AhmedBourouis/Scene-Sketch-Segmentation/HEAD/vpt/src/utils/logging.py --------------------------------------------------------------------------------