├── .gitignore ├── LICENSE ├── PerceptualSimilarity ├── data │ ├── __init__.py │ ├── base_data_loader.py │ ├── custom_dataset_data_loader.py │ ├── data_loader.py │ ├── dataset │ │ ├── __init__.py │ │ ├── base_dataset.py │ │ ├── jnd_dataset.py │ │ └── twoafc_dataset.py │ └── image_folder.py ├── imgs │ ├── ex_dir0 │ │ ├── 0.png │ │ └── 1.png │ ├── ex_dir1 │ │ ├── 0.png │ │ └── 1.png │ ├── ex_dir_pair │ │ ├── ex_p0.png │ │ ├── ex_p1.png │ │ └── ex_ref.png │ ├── ex_p0.png │ ├── ex_p1.png │ ├── ex_ref.png │ └── fig1.png ├── lpips │ ├── __init__.py │ ├── lpips.py │ ├── pretrained_networks.py │ ├── trainer.py │ └── weights │ │ ├── v0.0 │ │ ├── alex.pth │ │ ├── squeeze.pth │ │ └── vgg.pth │ │ └── v0.1 │ │ ├── alex.pth │ │ ├── squeeze.pth │ │ └── vgg.pth ├── lpips_1dir_allpairs.py ├── lpips_2dirs.py ├── lpips_2imgs.py ├── lpips_loss.py ├── scripts │ ├── download_dataset.sh │ ├── download_dataset_valonly.sh │ ├── eval_valsets.sh │ ├── train_test_metric.sh │ ├── train_test_metric_scratch.sh │ └── train_test_metric_tune.sh ├── setup.py ├── test_dataset_model.py ├── test_network.py ├── train.py └── util │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-34.pyc │ └── util.cpython-34.pyc │ ├── html.py │ ├── util.py │ └── visualizer.py ├── README.md ├── alexnet_eval ├── alexnet_quickdraw.py └── dataload.py ├── images ├── 2.png ├── 4.png └── overview.png ├── neuralline ├── jit.py ├── rasterize.py ├── rasterize_cuda.cpp └── rasterize_cuda_kernel.cu ├── requirements.txt ├── sample.py ├── sample.sh ├── setup.py ├── sketch_diffusion ├── Alexnet_acc.py ├── __init__.py ├── dist_util.py ├── fp16_util.py ├── gaussian_diffusion.py ├── gaussian_diffusion_acc.py ├── gaussian_diffusion_noise.py ├── image_datasets.py ├── logger.py ├── losses.py ├── nn.py ├── penet.py ├── resample.py ├── respace.py ├── script_util.py ├── train_util.py └── unet.py ├── train.py └── train.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/LICENSE -------------------------------------------------------------------------------- /PerceptualSimilarity/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PerceptualSimilarity/data/base_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/data/base_data_loader.py -------------------------------------------------------------------------------- /PerceptualSimilarity/data/custom_dataset_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/data/custom_dataset_data_loader.py -------------------------------------------------------------------------------- /PerceptualSimilarity/data/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/data/data_loader.py -------------------------------------------------------------------------------- /PerceptualSimilarity/data/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PerceptualSimilarity/data/dataset/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/data/dataset/base_dataset.py -------------------------------------------------------------------------------- /PerceptualSimilarity/data/dataset/jnd_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/data/dataset/jnd_dataset.py -------------------------------------------------------------------------------- /PerceptualSimilarity/data/dataset/twoafc_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/data/dataset/twoafc_dataset.py -------------------------------------------------------------------------------- /PerceptualSimilarity/data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/data/image_folder.py -------------------------------------------------------------------------------- /PerceptualSimilarity/imgs/ex_dir0/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/imgs/ex_dir0/0.png -------------------------------------------------------------------------------- /PerceptualSimilarity/imgs/ex_dir0/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/imgs/ex_dir0/1.png -------------------------------------------------------------------------------- /PerceptualSimilarity/imgs/ex_dir1/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/imgs/ex_dir1/0.png -------------------------------------------------------------------------------- /PerceptualSimilarity/imgs/ex_dir1/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/imgs/ex_dir1/1.png -------------------------------------------------------------------------------- /PerceptualSimilarity/imgs/ex_dir_pair/ex_p0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/imgs/ex_dir_pair/ex_p0.png -------------------------------------------------------------------------------- /PerceptualSimilarity/imgs/ex_dir_pair/ex_p1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/imgs/ex_dir_pair/ex_p1.png -------------------------------------------------------------------------------- /PerceptualSimilarity/imgs/ex_dir_pair/ex_ref.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/imgs/ex_dir_pair/ex_ref.png -------------------------------------------------------------------------------- /PerceptualSimilarity/imgs/ex_p0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/imgs/ex_p0.png -------------------------------------------------------------------------------- /PerceptualSimilarity/imgs/ex_p1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/imgs/ex_p1.png -------------------------------------------------------------------------------- /PerceptualSimilarity/imgs/ex_ref.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/imgs/ex_ref.png -------------------------------------------------------------------------------- /PerceptualSimilarity/imgs/fig1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/imgs/fig1.png -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips/__init__.py -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips/lpips.py -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips/pretrained_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips/pretrained_networks.py -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips/trainer.py -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips/weights/v0.0/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips/weights/v0.0/alex.pth -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips/weights/v0.0/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips/weights/v0.0/squeeze.pth -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips/weights/v0.0/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips/weights/v0.0/vgg.pth -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips/weights/v0.1/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips/weights/v0.1/alex.pth -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips/weights/v0.1/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips/weights/v0.1/squeeze.pth -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips/weights/v0.1/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips/weights/v0.1/vgg.pth -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips_1dir_allpairs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips_1dir_allpairs.py -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips_2dirs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips_2dirs.py -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips_2imgs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips_2imgs.py -------------------------------------------------------------------------------- /PerceptualSimilarity/lpips_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/lpips_loss.py -------------------------------------------------------------------------------- /PerceptualSimilarity/scripts/download_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/scripts/download_dataset.sh -------------------------------------------------------------------------------- /PerceptualSimilarity/scripts/download_dataset_valonly.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/scripts/download_dataset_valonly.sh -------------------------------------------------------------------------------- /PerceptualSimilarity/scripts/eval_valsets.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/scripts/eval_valsets.sh -------------------------------------------------------------------------------- /PerceptualSimilarity/scripts/train_test_metric.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/scripts/train_test_metric.sh -------------------------------------------------------------------------------- /PerceptualSimilarity/scripts/train_test_metric_scratch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/scripts/train_test_metric_scratch.sh -------------------------------------------------------------------------------- /PerceptualSimilarity/scripts/train_test_metric_tune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/scripts/train_test_metric_tune.sh -------------------------------------------------------------------------------- /PerceptualSimilarity/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/setup.py -------------------------------------------------------------------------------- /PerceptualSimilarity/test_dataset_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/test_dataset_model.py -------------------------------------------------------------------------------- /PerceptualSimilarity/test_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/test_network.py -------------------------------------------------------------------------------- /PerceptualSimilarity/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/train.py -------------------------------------------------------------------------------- /PerceptualSimilarity/util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PerceptualSimilarity/util/__pycache__/__init__.cpython-34.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/util/__pycache__/__init__.cpython-34.pyc -------------------------------------------------------------------------------- /PerceptualSimilarity/util/__pycache__/util.cpython-34.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/util/__pycache__/util.cpython-34.pyc -------------------------------------------------------------------------------- /PerceptualSimilarity/util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/util/html.py -------------------------------------------------------------------------------- /PerceptualSimilarity/util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/util/util.py -------------------------------------------------------------------------------- /PerceptualSimilarity/util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/PerceptualSimilarity/util/visualizer.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/README.md -------------------------------------------------------------------------------- /alexnet_eval/alexnet_quickdraw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/alexnet_eval/alexnet_quickdraw.py -------------------------------------------------------------------------------- /alexnet_eval/dataload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/alexnet_eval/dataload.py -------------------------------------------------------------------------------- /images/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/images/2.png -------------------------------------------------------------------------------- /images/4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/images/4.png -------------------------------------------------------------------------------- /images/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/images/overview.png -------------------------------------------------------------------------------- /neuralline/jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/neuralline/jit.py -------------------------------------------------------------------------------- /neuralline/rasterize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/neuralline/rasterize.py -------------------------------------------------------------------------------- /neuralline/rasterize_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/neuralline/rasterize_cuda.cpp -------------------------------------------------------------------------------- /neuralline/rasterize_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/neuralline/rasterize_cuda_kernel.cu -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/requirements.txt -------------------------------------------------------------------------------- /sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sample.py -------------------------------------------------------------------------------- /sample.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sample.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/setup.py -------------------------------------------------------------------------------- /sketch_diffusion/Alexnet_acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/Alexnet_acc.py -------------------------------------------------------------------------------- /sketch_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sketch_diffusion/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/dist_util.py -------------------------------------------------------------------------------- /sketch_diffusion/fp16_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/fp16_util.py -------------------------------------------------------------------------------- /sketch_diffusion/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/gaussian_diffusion.py -------------------------------------------------------------------------------- /sketch_diffusion/gaussian_diffusion_acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/gaussian_diffusion_acc.py -------------------------------------------------------------------------------- /sketch_diffusion/gaussian_diffusion_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/gaussian_diffusion_noise.py -------------------------------------------------------------------------------- /sketch_diffusion/image_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/image_datasets.py -------------------------------------------------------------------------------- /sketch_diffusion/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/logger.py -------------------------------------------------------------------------------- /sketch_diffusion/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/losses.py -------------------------------------------------------------------------------- /sketch_diffusion/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/nn.py -------------------------------------------------------------------------------- /sketch_diffusion/penet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/penet.py -------------------------------------------------------------------------------- /sketch_diffusion/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/resample.py -------------------------------------------------------------------------------- /sketch_diffusion/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/respace.py -------------------------------------------------------------------------------- /sketch_diffusion/script_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/script_util.py -------------------------------------------------------------------------------- /sketch_diffusion/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/train_util.py -------------------------------------------------------------------------------- /sketch_diffusion/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/sketch_diffusion/unet.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangqiang9/SketchKnitter/HEAD/train.sh --------------------------------------------------------------------------------