From 5cb1f588525679c4f02b0d58208fe4aa2349d535 Mon Sep 17 00:00:00 2001 From: areszz <1031614818@qq.com> Date: Sat, 22 Feb 2025 14:21:54 +0800 Subject: [PATCH] first commit --- README.md | 142 + data/__init__.py | 98 + data/__pycache__/__init__.cpython-36.pyc | Bin 0 -> 4139 bytes data/__pycache__/base_dataset.cpython-36.pyc | Bin 0 -> 8090 bytes data/__pycache__/image_folder.cpython-36.pyc | Bin 0 -> 2469 bytes .../unaligned_dataset.cpython-36.pyc | Bin 0 -> 3052 bytes .../unaligned_double_dataset.cpython-36.pyc | Bin 0 -> 3634 bytes data/base_dataset.py | 230 ++ data/image_folder.py | 66 + data/single_dataset.py | 40 + data/singleimage_dataset.py | 108 + data/template_dataset.py | 75 + data/unaligned_dataset.py | 79 + data/unaligned_double_dataset.py | 100 + datasets/bibtex/cityscapes.tex | 6 + datasets/bibtex/facades.tex | 7 + datasets/bibtex/handbags.tex | 13 + datasets/bibtex/shoes.tex | 14 + datasets/bibtex/transattr.tex | 8 + datasets/combine_A_and_B.py | 48 + datasets/detect_cat_face.py | 64 + datasets/download_cut_dataset.sh | 23 + datasets/download_pix2pix_dataset.sh | 24 + datasets/make_dataset_aligned.py | 63 + datasets/prepare_cityscapes_dataset.py | 90 + .../trainA/monet.jpg | Bin 0 -> 296150 bytes .../trainB/etretat-normandy-france.jpg | Bin 0 -> 620384 bytes images/method_final.jpg | Bin 0 -> 284293 bytes models/__init__.py | 67 + models/__pycache__/__init__.cpython-36.pyc | Bin 0 -> 3220 bytes models/__pycache__/base_model.cpython-36.pyc | Bin 0 -> 11154 bytes models/__pycache__/cut_model.cpython-36.pyc | Bin 0 -> 8069 bytes models/__pycache__/mae.cpython-36.pyc | Bin 0 -> 8422 bytes models/__pycache__/models_mae.cpython-36.pyc | Bin 0 -> 8344 bytes .../mutilvitgloballocal_model.cpython-36.pyc | Bin 0 -> 11463 bytes models/__pycache__/networks.cpython-36.pyc | Bin 0 -> 52658 bytes models/__pycache__/patchnce.cpython-36.pyc | Bin 0 -> 1568 bytes .../__pycache__/region0_model.cpython-36.pyc | Bin 0 -> 12992 bytes .../__pycache__/region_model.cpython-36.pyc | Bin 0 -> 12018 bytes .../stylegan_networks.cpython-36.pyc | Bin 0 -> 23294 bytes .../vit2Gmask_model.cpython-36.pyc | Bin 0 -> 9239 bytes models/__pycache__/vit2_model.cpython-36.pyc | Bin 0 -> 9372 bytes .../vit2patchmask_model.cpython-36.pyc | Bin 0 -> 9713 bytes .../vit2tokenmask_model.cpython-36.pyc | Bin 0 -> 9189 bytes models/__pycache__/vitD_model.cpython-36.pyc | Bin 0 -> 9777 bytes models/__pycache__/vit_model.cpython-36.pyc | Bin 0 -> 8707 bytes .../vitdonly2_model.cpython-36.pyc | Bin 0 -> 9528 bytes .../__pycache__/vitdonly_model.cpython-36.pyc | Bin 0 -> 9373 bytes .../vitgloballocal_model.cpython-36.pyc | Bin 0 -> 11152 bytes .../vitlocalgloballocal_model.cpython-36.pyc | Bin 0 -> 10505 bytes models/base_model.py | 258 ++ models/cut_model.py | 214 ++ models/cycle_gan_model.py | 222 ++ models/networks.py | 1530 +++++++++++ models/patchnce.py | 55 + models/roma_model.py | 363 +++ models/roma_single_model.py | 272 ++ models/self_build.py | 655 +++++ models/stylegan_networks.py | 914 +++++++ models/template_model.py | 99 + .../util/__pycache__/pos_embed.cpython-36.pyc | Bin 0 -> 2424 bytes models/util/crop.py | 42 + models/util/datasets.py | 65 + models/util/lars.py | 47 + models/util/lr_decay.py | 76 + models/util/lr_sched.py | 21 + models/util/misc.py | 340 +++ models/util/pos_embed.py | 96 + options/__init__.py | 1 + options/__pycache__/__init__.cpython-36.pyc | Bin 0 -> 292 bytes .../__pycache__/base_options.cpython-36.pyc | Bin 0 -> 7539 bytes .../__pycache__/test_options.cpython-36.pyc | Bin 0 -> 1039 bytes .../__pycache__/train_options.cpython-36.pyc | Bin 0 -> 3174 bytes options/base_options.py | 167 ++ options/test_options.py | 21 + options/train_options.py | 47 + scripts/test.sh | 1 + scripts/train.sh | 5 + test.py | 70 + timm/__init__.py | 4 + timm/__pycache__/__init__.cpython-36.pyc | Bin 0 -> 550 bytes timm/__pycache__/version.cpython-36.pyc | Bin 0 -> 168 bytes timm/data/__init__.py | 12 + timm/data/__pycache__/__init__.cpython-36.pyc | Bin 0 -> 861 bytes .../__pycache__/auto_augment.cpython-36.pyc | Bin 0 -> 27337 bytes timm/data/__pycache__/config.cpython-36.pyc | Bin 0 -> 1609 bytes .../data/__pycache__/constants.cpython-36.pyc | Bin 0 -> 598 bytes timm/data/__pycache__/dataset.cpython-36.pyc | Bin 0 -> 5013 bytes .../dataset_factory.cpython-36.pyc | Bin 0 -> 4141 bytes .../distributed_sampler.cpython-36.pyc | Bin 0 -> 4229 bytes timm/data/__pycache__/loader.cpython-36.pyc | Bin 0 -> 7910 bytes timm/data/__pycache__/mixup.cpython-36.pyc | Bin 0 -> 11571 bytes .../__pycache__/random_erasing.cpython-36.pyc | Bin 0 -> 3940 bytes .../__pycache__/real_labels.cpython-36.pyc | Bin 0 -> 2390 bytes .../__pycache__/transforms.cpython-36.pyc | Bin 0 -> 6434 bytes .../transforms_factory.cpython-36.pyc | Bin 0 -> 5119 bytes timm/data/auto_augment.py | 865 +++++++ timm/data/config.py | 78 + timm/data/constants.py | 7 + timm/data/dataset.py | 152 ++ timm/data/dataset_factory.py | 139 + timm/data/distributed_sampler.py | 128 + timm/data/loader.py | 289 +++ timm/data/mixup.py | 316 +++ timm/data/parsers/__init__.py | 1 + .../__pycache__/__init__.cpython-36.pyc | Bin 0 -> 213 bytes .../__pycache__/class_map.cpython-36.pyc | Bin 0 -> 934 bytes .../__pycache__/constants.cpython-36.pyc | Bin 0 -> 215 bytes .../parsers/__pycache__/parser.cpython-36.pyc | Bin 0 -> 1134 bytes .../__pycache__/parser_factory.cpython-36.pyc | Bin 0 -> 874 bytes .../parser_image_folder.cpython-36.pyc | Bin 0 -> 3026 bytes .../parser_image_in_tar.cpython-36.pyc | Bin 0 -> 7672 bytes .../parser_image_tar.cpython-36.pyc | Bin 0 -> 3104 bytes timm/data/parsers/class_map.py | 19 + timm/data/parsers/constants.py | 1 + timm/data/parsers/parser.py | 17 + timm/data/parsers/parser_factory.py | 29 + timm/data/parsers/parser_image_folder.py | 69 + timm/data/parsers/parser_image_in_tar.py | 222 ++ timm/data/parsers/parser_image_tar.py | 72 + timm/data/parsers/parser_tfds.py | 297 +++ timm/data/random_erasing.py | 103 + timm/data/real_labels.py | 42 + timm/data/tf_preprocessing.py | 232 ++ timm/data/transforms.py | 185 ++ timm/data/transforms_factory.py | 236 ++ timm/loss/__init__.py | 4 + timm/loss/asymmetric_loss.py | 97 + timm/loss/binary_cross_entropy.py | 47 + timm/loss/cross_entropy.py | 36 + timm/loss/jsd.py | 39 + timm/models/__init__.py | 58 + .../__pycache__/__init__.cpython-36.pyc | Bin 0 -> 1929 bytes timm/models/__pycache__/beit.cpython-36.pyc | Bin 0 -> 13767 bytes .../models/__pycache__/byoanet.cpython-36.pyc | Bin 0 -> 11525 bytes .../models/__pycache__/byobnet.cpython-36.pyc | Bin 0 -> 43550 bytes timm/models/__pycache__/cait.cpython-36.pyc | Bin 0 -> 13252 bytes timm/models/__pycache__/coat.cpython-36.pyc | Bin 0 -> 20181 bytes timm/models/__pycache__/convit.cpython-36.pyc | Bin 0 -> 12239 bytes .../__pycache__/convmixer.cpython-36.pyc | Bin 0 -> 4159 bytes .../__pycache__/crossvit.cpython-36.pyc | Bin 0 -> 19691 bytes timm/models/__pycache__/cspnet.cpython-36.pyc | Bin 0 -> 14814 bytes .../__pycache__/densenet.cpython-36.pyc | Bin 0 -> 13075 bytes timm/models/__pycache__/dla.cpython-36.pyc | Bin 0 -> 14562 bytes timm/models/__pycache__/dpn.cpython-36.pyc | Bin 0 -> 10582 bytes .../__pycache__/efficientnet.cpython-36.pyc | Bin 0 -> 75391 bytes .../efficientnet_blocks.cpython-36.pyc | Bin 0 -> 9769 bytes .../efficientnet_builder.cpython-36.pyc | Bin 0 -> 12660 bytes .../models/__pycache__/factory.cpython-36.pyc | Bin 0 -> 3023 bytes .../__pycache__/features.cpython-36.pyc | Bin 0 -> 12545 bytes .../__pycache__/fx_features.cpython-36.pyc | Bin 0 -> 2660 bytes .../__pycache__/ghostnet.cpython-36.pyc | Bin 0 -> 7532 bytes .../__pycache__/gluon_resnet.cpython-36.pyc | Bin 0 -> 10128 bytes .../__pycache__/gluon_xception.cpython-36.pyc | Bin 0 -> 6881 bytes .../__pycache__/hardcorenas.cpython-36.pyc | Bin 0 -> 6386 bytes .../models/__pycache__/helpers.cpython-36.pyc | Bin 0 -> 15038 bytes timm/models/__pycache__/hrnet.cpython-36.pyc | Bin 0 -> 19111 bytes timm/models/__pycache__/hub.cpython-36.pyc | Bin 0 -> 5289 bytes .../inception_resnet_v2.cpython-36.pyc | Bin 0 -> 10718 bytes .../__pycache__/inception_v3.cpython-36.pyc | Bin 0 -> 14117 bytes .../__pycache__/inception_v4.cpython-36.pyc | Bin 0 -> 10692 bytes timm/models/__pycache__/levit.cpython-36.pyc | Bin 0 -> 17526 bytes .../__pycache__/mlp_mixer.cpython-36.pyc | Bin 0 -> 23914 bytes .../__pycache__/mobilenetv3.cpython-36.pyc | Bin 0 -> 16609 bytes timm/models/__pycache__/nasnet.cpython-36.pyc | Bin 0 -> 15831 bytes timm/models/__pycache__/nest.cpython-36.pyc | Bin 0 -> 18311 bytes timm/models/__pycache__/nfnet.cpython-36.pyc | Bin 0 -> 33335 bytes timm/models/__pycache__/pit.cpython-36.pyc | Bin 0 -> 12341 bytes .../models/__pycache__/pnasnet.cpython-36.pyc | Bin 0 -> 10870 bytes .../__pycache__/registry.cpython-36.pyc | Bin 0 -> 4729 bytes timm/models/__pycache__/regnet.cpython-36.pyc | Bin 0 -> 18516 bytes .../models/__pycache__/res2net.cpython-36.pyc | Bin 0 -> 6980 bytes .../models/__pycache__/resnest.cpython-36.pyc | Bin 0 -> 8441 bytes timm/models/__pycache__/resnet.cpython-36.pyc | Bin 0 -> 54345 bytes .../__pycache__/resnetv2.cpython-36.pyc | Bin 0 -> 23075 bytes timm/models/__pycache__/rexnet.cpython-36.pyc | Bin 0 -> 8779 bytes .../__pycache__/selecsls.cpython-36.pyc | Bin 0 -> 10933 bytes timm/models/__pycache__/senet.cpython-36.pyc | Bin 0 -> 15147 bytes timm/models/__pycache__/sknet.cpython-36.pyc | Bin 0 -> 7739 bytes .../swin_transformer.cpython-36.pyc | Bin 0 -> 23074 bytes timm/models/__pycache__/tnt.cpython-36.pyc | Bin 0 -> 9905 bytes .../models/__pycache__/tresnet.cpython-36.pyc | Bin 0 -> 9691 bytes timm/models/__pycache__/twins.cpython-36.pyc | Bin 0 -> 15148 bytes timm/models/__pycache__/vgg.cpython-36.pyc | Bin 0 -> 9627 bytes .../__pycache__/visformer.cpython-36.pyc | Bin 0 -> 11099 bytes .../vision_transformer.cpython-36.pyc | Bin 0 -> 38076 bytes .../vision_transformer_hybrid.cpython-36.pyc | Bin 0 -> 13017 bytes timm/models/__pycache__/vovnet.cpython-36.pyc | Bin 0 -> 10478 bytes .../__pycache__/xception.cpython-36.pyc | Bin 0 -> 6618 bytes .../xception_aligned.cpython-36.pyc | Bin 0 -> 7647 bytes timm/models/__pycache__/xcit.cpython-36.pyc | Bin 0 -> 31091 bytes timm/models/beit.py | 416 +++ timm/models/byoanet.py | 443 ++++ timm/models/byobnet.py | 1531 +++++++++++ timm/models/cait.py | 394 +++ timm/models/coat.py | 661 +++++ timm/models/convit.py | 351 +++ timm/models/convmixer.py | 101 + timm/models/crossvit.py | 517 ++++ timm/models/cspnet.py | 457 ++++ timm/models/densenet.py | 387 +++ timm/models/dla.py | 443 ++++ timm/models/dpn.py | 317 +++ timm/models/efficientnet.py | 2286 +++++++++++++++++ timm/models/efficientnet_blocks.py | 323 +++ timm/models/efficientnet_builder.py | 463 ++++ timm/models/factory.py | 86 + timm/models/features.py | 284 ++ timm/models/fx_features.py | 73 + timm/models/ghostnet.py | 276 ++ timm/models/gluon_resnet.py | 248 ++ timm/models/gluon_xception.py | 246 ++ timm/models/hardcorenas.py | 152 ++ timm/models/helpers.py | 518 ++++ timm/models/hrnet.py | 836 ++++++ timm/models/hub.py | 171 ++ timm/models/inception_resnet_v2.py | 358 +++ timm/models/inception_v3.py | 470 ++++ timm/models/inception_v4.py | 316 +++ timm/models/layers/__init__.py | 40 + .../__pycache__/__init__.cpython-36.pyc | Bin 0 -> 3100 bytes .../__pycache__/activations.cpython-36.pyc | Bin 0 -> 6831 bytes .../activations_jit.cpython-36.pyc | Bin 0 -> 4303 bytes .../__pycache__/activations_me.cpython-36.pyc | Bin 0 -> 9158 bytes .../adaptive_avgmax_pool.cpython-36.pyc | Bin 0 -> 4793 bytes .../__pycache__/blur_pool.cpython-36.pyc | Bin 0 -> 2072 bytes .../bottleneck_attn.cpython-36.pyc | Bin 0 -> 6437 bytes .../layers/__pycache__/cbam.cpython-36.pyc | Bin 0 -> 5400 bytes .../__pycache__/classifier.cpython-36.pyc | Bin 0 -> 2200 bytes .../__pycache__/cond_conv2d.cpython-36.pyc | Bin 0 -> 3761 bytes .../layers/__pycache__/config.cpython-36.pyc | Bin 0 -> 3457 bytes .../__pycache__/conv2d_same.cpython-36.pyc | Bin 0 -> 1899 bytes .../__pycache__/conv_bn_act.cpython-36.pyc | Bin 0 -> 1588 bytes .../__pycache__/create_act.cpython-36.pyc | Bin 0 -> 3650 bytes .../__pycache__/create_attn.cpython-36.pyc | Bin 0 -> 1968 bytes .../__pycache__/create_conv2d.cpython-36.pyc | Bin 0 -> 1065 bytes .../create_norm_act.cpython-36.pyc | Bin 0 -> 2310 bytes .../layers/__pycache__/drop.cpython-36.pyc | Bin 0 -> 5677 bytes .../layers/__pycache__/eca.cpython-36.pyc | Bin 0 -> 6178 bytes .../__pycache__/evo_norm.cpython-36.pyc | Bin 0 -> 3610 bytes .../__pycache__/gather_excite.cpython-36.pyc | Bin 0 -> 3076 bytes .../__pycache__/global_context.cpython-36.pyc | Bin 0 -> 2402 bytes .../__pycache__/halo_attn.cpython-36.pyc | Bin 0 -> 7544 bytes .../layers/__pycache__/helpers.cpython-36.pyc | Bin 0 -> 998 bytes .../__pycache__/inplace_abn.cpython-36.pyc | Bin 0 -> 3124 bytes .../__pycache__/lambda_layer.cpython-36.pyc | Bin 0 -> 5497 bytes .../layers/__pycache__/linear.cpython-36.pyc | Bin 0 -> 1050 bytes .../__pycache__/mixed_conv2d.cpython-36.pyc | Bin 0 -> 2226 bytes .../layers/__pycache__/mlp.cpython-36.pyc | Bin 0 -> 4076 bytes .../__pycache__/non_local_attn.cpython-36.pyc | Bin 0 -> 5634 bytes .../layers/__pycache__/norm.cpython-36.pyc | Bin 0 -> 1539 bytes .../__pycache__/norm_act.cpython-36.pyc | Bin 0 -> 3010 bytes .../layers/__pycache__/padding.cpython-36.pyc | Bin 0 -> 1767 bytes .../__pycache__/patch_embed.cpython-36.pyc | Bin 0 -> 1711 bytes .../__pycache__/pool2d_same.cpython-36.pyc | Bin 0 -> 3168 bytes .../selective_kernel.cpython-36.pyc | Bin 0 -> 5532 bytes .../__pycache__/separable_conv.cpython-36.pyc | Bin 0 -> 2910 bytes .../__pycache__/space_to_depth.cpython-36.pyc | Bin 0 -> 2532 bytes .../__pycache__/split_attn.cpython-36.pyc | Bin 0 -> 2956 bytes .../split_batchnorm.cpython-36.pyc | Bin 0 -> 3356 bytes .../__pycache__/squeeze_excite.cpython-36.pyc | Bin 0 -> 3251 bytes .../__pycache__/std_conv.cpython-36.pyc | Bin 0 -> 6033 bytes .../__pycache__/test_time_pool.cpython-36.pyc | Bin 0 -> 2091 bytes .../__pycache__/trace_utils.cpython-36.pyc | Bin 0 -> 684 bytes .../__pycache__/weight_init.cpython-36.pyc | Bin 0 -> 2669 bytes timm/models/layers/activations.py | 145 ++ timm/models/layers/activations_jit.py | 90 + timm/models/layers/activations_me.py | 218 ++ timm/models/layers/adaptive_avgmax_pool.py | 118 + timm/models/layers/attention_pool2d.py | 182 ++ timm/models/layers/blur_pool.py | 42 + timm/models/layers/bottleneck_attn.py | 157 ++ timm/models/layers/cbam.py | 112 + timm/models/layers/classifier.py | 56 + timm/models/layers/cond_conv2d.py | 122 + timm/models/layers/config.py | 115 + timm/models/layers/conv2d_same.py | 42 + timm/models/layers/conv_bn_act.py | 40 + timm/models/layers/create_act.py | 153 ++ timm/models/layers/create_attn.py | 89 + timm/models/layers/create_conv2d.py | 31 + timm/models/layers/create_norm_act.py | 83 + timm/models/layers/drop.py | 168 ++ timm/models/layers/eca.py | 145 ++ timm/models/layers/evo_norm.py | 81 + timm/models/layers/gather_excite.py | 90 + timm/models/layers/global_context.py | 67 + timm/models/layers/halo_attn.py | 233 ++ timm/models/layers/helpers.py | 31 + timm/models/layers/inplace_abn.py | 87 + timm/models/layers/lambda_layer.py | 133 + timm/models/layers/linear.py | 19 + timm/models/layers/median_pool.py | 49 + timm/models/layers/mixed_conv2d.py | 51 + timm/models/layers/mlp.py | 119 + timm/models/layers/non_local_attn.py | 145 ++ timm/models/layers/norm.py | 24 + timm/models/layers/norm_act.py | 85 + timm/models/layers/padding.py | 56 + timm/models/layers/patch_embed.py | 39 + timm/models/layers/pool2d_same.py | 73 + timm/models/layers/selective_kernel.py | 120 + timm/models/layers/separable_conv.py | 73 + timm/models/layers/space_to_depth.py | 53 + timm/models/layers/split_attn.py | 85 + timm/models/layers/split_batchnorm.py | 75 + timm/models/layers/squeeze_excite.py | 74 + timm/models/layers/std_conv.py | 133 + timm/models/layers/test_time_pool.py | 52 + timm/models/layers/trace_utils.py | 13 + timm/models/layers/weight_init.py | 89 + timm/models/levit.py | 563 ++++ timm/models/mlp_mixer.py | 659 +++++ timm/models/mobilenetv3.py | 562 ++++ timm/models/nasnet.py | 567 ++++ timm/models/nest.py | 465 ++++ timm/models/nfnet.py | 968 +++++++ timm/models/pit.py | 384 +++ timm/models/pnasnet.py | 350 +++ timm/models/pruned/ecaresnet101d_pruned.txt | 1 + timm/models/pruned/ecaresnet50d_pruned.txt | 1 + timm/models/pruned/efficientnet_b1_pruned.txt | 1 + timm/models/pruned/efficientnet_b2_pruned.txt | 1 + timm/models/pruned/efficientnet_b3_pruned.txt | 1 + timm/models/registry.py | 149 ++ timm/models/regnet.py | 494 ++++ timm/models/res2net.py | 216 ++ timm/models/resnest.py | 237 ++ timm/models/resnet.py | 1472 +++++++++++ timm/models/resnetv2.py | 672 +++++ timm/models/rexnet.py | 239 ++ timm/models/selecsls.py | 362 +++ timm/models/senet.py | 467 ++++ timm/models/sknet.py | 215 ++ timm/models/swin_transformer.py | 656 +++++ timm/models/tnt.py | 272 ++ timm/models/tresnet.py | 297 +++ timm/models/twins.py | 424 +++ timm/models/vgg.py | 263 ++ timm/models/visformer.py | 412 +++ timm/models/vision_transformer.py | 989 +++++++ timm/models/vision_transformer_hybrid.py | 363 +++ timm/models/vovnet.py | 406 +++ timm/models/xception.py | 232 ++ timm/models/xception_aligned.py | 238 ++ timm/models/xcit.py | 812 ++++++ timm/optim/__init__.py | 15 + timm/optim/adabelief.py | 201 ++ timm/optim/adafactor.py | 167 ++ timm/optim/adahessian.py | 156 ++ timm/optim/adamp.py | 105 + timm/optim/adamw.py | 122 + timm/optim/lamb.py | 192 ++ timm/optim/lars.py | 135 + timm/optim/lookahead.py | 61 + timm/optim/madgrad.py | 184 ++ timm/optim/nadam.py | 92 + timm/optim/nvnovograd.py | 120 + timm/optim/optim_factory.py | 217 ++ timm/optim/radam.py | 89 + timm/optim/rmsprop_tf.py | 139 + timm/optim/sgdp.py | 70 + timm/scheduler/__init__.py | 8 + timm/scheduler/cosine_lr.py | 119 + timm/scheduler/multistep_lr.py | 65 + timm/scheduler/plateau_lr.py | 113 + timm/scheduler/poly_lr.py | 116 + timm/scheduler/scheduler.py | 105 + timm/scheduler/scheduler_factory.py | 107 + timm/scheduler/step_lr.py | 63 + timm/scheduler/tanh_lr.py | 117 + timm/utils/__init__.py | 13 + .../utils/__pycache__/__init__.cpython-36.pyc | Bin 0 -> 961 bytes timm/utils/__pycache__/agc.cpython-36.pyc | Bin 0 -> 1526 bytes .../checkpoint_saver.cpython-36.pyc | Bin 0 -> 4431 bytes .../__pycache__/clip_grad.cpython-36.pyc | Bin 0 -> 960 bytes timm/utils/__pycache__/cuda.cpython-36.pyc | Bin 0 -> 2215 bytes .../__pycache__/distributed.cpython-36.pyc | Bin 0 -> 939 bytes timm/utils/__pycache__/jit.cpython-36.pyc | Bin 0 -> 810 bytes timm/utils/__pycache__/log.cpython-36.pyc | Bin 0 -> 1369 bytes timm/utils/__pycache__/metrics.cpython-36.pyc | Bin 0 -> 1610 bytes timm/utils/__pycache__/misc.cpython-36.pyc | Bin 0 -> 1006 bytes timm/utils/__pycache__/model.cpython-36.pyc | Bin 0 -> 11239 bytes .../__pycache__/model_ema.cpython-36.pyc | Bin 0 -> 5801 bytes timm/utils/__pycache__/random.cpython-36.pyc | Bin 0 -> 416 bytes timm/utils/__pycache__/summary.cpython-36.pyc | Bin 0 -> 1487 bytes timm/utils/agc.py | 42 + timm/utils/checkpoint_saver.py | 150 ++ timm/utils/clip_grad.py | 23 + timm/utils/cuda.py | 55 + timm/utils/distributed.py | 28 + timm/utils/jit.py | 18 + timm/utils/log.py | 28 + timm/utils/metrics.py | 32 + timm/utils/misc.py | 18 + timm/utils/model.py | 273 ++ timm/utils/model_ema.py | 126 + timm/utils/random.py | 9 + timm/utils/summary.py | 39 + timm/version.py | 1 + train.py | 77 + util/__init__.py | 2 + util/__pycache__/__init__.cpython-36.pyc | Bin 0 -> 262 bytes util/__pycache__/html.cpython-36.pyc | Bin 0 -> 3567 bytes util/__pycache__/util.cpython-36.pyc | Bin 0 -> 5239 bytes util/__pycache__/visualizer.cpython-36.pyc | Bin 0 -> 8660 bytes util/get_data.py | 110 + util/html.py | 86 + util/image_pool.py | 54 + util/util.py | 166 ++ util/visualizer.py | 242 ++ 411 files changed, 47043 insertions(+) create mode 100644 README.md create mode 100644 data/__init__.py create mode 100644 data/__pycache__/__init__.cpython-36.pyc create mode 100644 data/__pycache__/base_dataset.cpython-36.pyc create mode 100644 data/__pycache__/image_folder.cpython-36.pyc create mode 100644 data/__pycache__/unaligned_dataset.cpython-36.pyc create mode 100644 data/__pycache__/unaligned_double_dataset.cpython-36.pyc create mode 100644 data/base_dataset.py create mode 100644 data/image_folder.py create mode 100644 data/single_dataset.py create mode 100644 data/singleimage_dataset.py create mode 100644 data/template_dataset.py create mode 100644 data/unaligned_dataset.py create mode 100644 data/unaligned_double_dataset.py create mode 100644 datasets/bibtex/cityscapes.tex create mode 100644 datasets/bibtex/facades.tex create mode 100644 datasets/bibtex/handbags.tex create mode 100644 datasets/bibtex/shoes.tex create mode 100644 datasets/bibtex/transattr.tex create mode 100644 datasets/combine_A_and_B.py create mode 100644 datasets/detect_cat_face.py create mode 100644 datasets/download_cut_dataset.sh create mode 100644 datasets/download_pix2pix_dataset.sh create mode 100644 datasets/make_dataset_aligned.py create mode 100644 datasets/prepare_cityscapes_dataset.py create mode 100644 datasets/single_image_monet_etretat/trainA/monet.jpg create mode 100644 datasets/single_image_monet_etretat/trainB/etretat-normandy-france.jpg create mode 100644 images/method_final.jpg create mode 100644 models/__init__.py create mode 100644 models/__pycache__/__init__.cpython-36.pyc create mode 100644 models/__pycache__/base_model.cpython-36.pyc create mode 100644 models/__pycache__/cut_model.cpython-36.pyc create mode 100644 models/__pycache__/mae.cpython-36.pyc create mode 100644 models/__pycache__/models_mae.cpython-36.pyc create mode 100644 models/__pycache__/mutilvitgloballocal_model.cpython-36.pyc create mode 100644 models/__pycache__/networks.cpython-36.pyc create mode 100644 models/__pycache__/patchnce.cpython-36.pyc create mode 100644 models/__pycache__/region0_model.cpython-36.pyc create mode 100644 models/__pycache__/region_model.cpython-36.pyc create mode 100644 models/__pycache__/stylegan_networks.cpython-36.pyc create mode 100644 models/__pycache__/vit2Gmask_model.cpython-36.pyc create mode 100644 models/__pycache__/vit2_model.cpython-36.pyc create mode 100644 models/__pycache__/vit2patchmask_model.cpython-36.pyc create mode 100644 models/__pycache__/vit2tokenmask_model.cpython-36.pyc create mode 100644 models/__pycache__/vitD_model.cpython-36.pyc create mode 100644 models/__pycache__/vit_model.cpython-36.pyc create mode 100644 models/__pycache__/vitdonly2_model.cpython-36.pyc create mode 100644 models/__pycache__/vitdonly_model.cpython-36.pyc create mode 100644 models/__pycache__/vitgloballocal_model.cpython-36.pyc create mode 100644 models/__pycache__/vitlocalgloballocal_model.cpython-36.pyc create mode 100644 models/base_model.py create mode 100644 models/cut_model.py create mode 100644 models/cycle_gan_model.py create mode 100644 models/networks.py create mode 100644 models/patchnce.py create mode 100644 models/roma_model.py create mode 100644 models/roma_single_model.py create mode 100644 models/self_build.py create mode 100644 models/stylegan_networks.py create mode 100644 models/template_model.py create mode 100644 models/util/__pycache__/pos_embed.cpython-36.pyc create mode 100644 models/util/crop.py create mode 100644 models/util/datasets.py create mode 100644 models/util/lars.py create mode 100644 models/util/lr_decay.py create mode 100644 models/util/lr_sched.py create mode 100644 models/util/misc.py create mode 100644 models/util/pos_embed.py create mode 100644 options/__init__.py create mode 100644 options/__pycache__/__init__.cpython-36.pyc create mode 100644 options/__pycache__/base_options.cpython-36.pyc create mode 100644 options/__pycache__/test_options.cpython-36.pyc create mode 100644 options/__pycache__/train_options.cpython-36.pyc create mode 100644 options/base_options.py create mode 100644 options/test_options.py create mode 100644 options/train_options.py create mode 100644 scripts/test.sh create mode 100644 scripts/train.sh create mode 100644 test.py create mode 100644 timm/__init__.py create mode 100644 timm/__pycache__/__init__.cpython-36.pyc create mode 100644 timm/__pycache__/version.cpython-36.pyc create mode 100644 timm/data/__init__.py create mode 100644 timm/data/__pycache__/__init__.cpython-36.pyc create mode 100644 timm/data/__pycache__/auto_augment.cpython-36.pyc create mode 100644 timm/data/__pycache__/config.cpython-36.pyc create mode 100644 timm/data/__pycache__/constants.cpython-36.pyc create mode 100644 timm/data/__pycache__/dataset.cpython-36.pyc create mode 100644 timm/data/__pycache__/dataset_factory.cpython-36.pyc create mode 100644 timm/data/__pycache__/distributed_sampler.cpython-36.pyc create mode 100644 timm/data/__pycache__/loader.cpython-36.pyc create mode 100644 timm/data/__pycache__/mixup.cpython-36.pyc create mode 100644 timm/data/__pycache__/random_erasing.cpython-36.pyc create mode 100644 timm/data/__pycache__/real_labels.cpython-36.pyc create mode 100644 timm/data/__pycache__/transforms.cpython-36.pyc create mode 100644 timm/data/__pycache__/transforms_factory.cpython-36.pyc create mode 100644 timm/data/auto_augment.py create mode 100644 timm/data/config.py create mode 100644 timm/data/constants.py create mode 100644 timm/data/dataset.py create mode 100644 timm/data/dataset_factory.py create mode 100644 timm/data/distributed_sampler.py create mode 100644 timm/data/loader.py create mode 100644 timm/data/mixup.py create mode 100644 timm/data/parsers/__init__.py create mode 100644 timm/data/parsers/__pycache__/__init__.cpython-36.pyc create mode 100644 timm/data/parsers/__pycache__/class_map.cpython-36.pyc create mode 100644 timm/data/parsers/__pycache__/constants.cpython-36.pyc create mode 100644 timm/data/parsers/__pycache__/parser.cpython-36.pyc create mode 100644 timm/data/parsers/__pycache__/parser_factory.cpython-36.pyc create mode 100644 timm/data/parsers/__pycache__/parser_image_folder.cpython-36.pyc create mode 100644 timm/data/parsers/__pycache__/parser_image_in_tar.cpython-36.pyc create mode 100644 timm/data/parsers/__pycache__/parser_image_tar.cpython-36.pyc create mode 100644 timm/data/parsers/class_map.py create mode 100644 timm/data/parsers/constants.py create mode 100644 timm/data/parsers/parser.py create mode 100644 timm/data/parsers/parser_factory.py create mode 100644 timm/data/parsers/parser_image_folder.py create mode 100644 timm/data/parsers/parser_image_in_tar.py create mode 100644 timm/data/parsers/parser_image_tar.py create mode 100644 timm/data/parsers/parser_tfds.py create mode 100644 timm/data/random_erasing.py create mode 100644 timm/data/real_labels.py create mode 100644 timm/data/tf_preprocessing.py create mode 100644 timm/data/transforms.py create mode 100644 timm/data/transforms_factory.py create mode 100644 timm/loss/__init__.py create mode 100644 timm/loss/asymmetric_loss.py create mode 100644 timm/loss/binary_cross_entropy.py create mode 100644 timm/loss/cross_entropy.py create mode 100644 timm/loss/jsd.py create mode 100644 timm/models/__init__.py create mode 100644 timm/models/__pycache__/__init__.cpython-36.pyc create mode 100644 timm/models/__pycache__/beit.cpython-36.pyc create mode 100644 timm/models/__pycache__/byoanet.cpython-36.pyc create mode 100644 timm/models/__pycache__/byobnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/cait.cpython-36.pyc create mode 100644 timm/models/__pycache__/coat.cpython-36.pyc create mode 100644 timm/models/__pycache__/convit.cpython-36.pyc create mode 100644 timm/models/__pycache__/convmixer.cpython-36.pyc create mode 100644 timm/models/__pycache__/crossvit.cpython-36.pyc create mode 100644 timm/models/__pycache__/cspnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/densenet.cpython-36.pyc create mode 100644 timm/models/__pycache__/dla.cpython-36.pyc create mode 100644 timm/models/__pycache__/dpn.cpython-36.pyc create mode 100644 timm/models/__pycache__/efficientnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/efficientnet_blocks.cpython-36.pyc create mode 100644 timm/models/__pycache__/efficientnet_builder.cpython-36.pyc create mode 100644 timm/models/__pycache__/factory.cpython-36.pyc create mode 100644 timm/models/__pycache__/features.cpython-36.pyc create mode 100644 timm/models/__pycache__/fx_features.cpython-36.pyc create mode 100644 timm/models/__pycache__/ghostnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/gluon_resnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/gluon_xception.cpython-36.pyc create mode 100644 timm/models/__pycache__/hardcorenas.cpython-36.pyc create mode 100644 timm/models/__pycache__/helpers.cpython-36.pyc create mode 100644 timm/models/__pycache__/hrnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/hub.cpython-36.pyc create mode 100644 timm/models/__pycache__/inception_resnet_v2.cpython-36.pyc create mode 100644 timm/models/__pycache__/inception_v3.cpython-36.pyc create mode 100644 timm/models/__pycache__/inception_v4.cpython-36.pyc create mode 100644 timm/models/__pycache__/levit.cpython-36.pyc create mode 100644 timm/models/__pycache__/mlp_mixer.cpython-36.pyc create mode 100644 timm/models/__pycache__/mobilenetv3.cpython-36.pyc create mode 100644 timm/models/__pycache__/nasnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/nest.cpython-36.pyc create mode 100644 timm/models/__pycache__/nfnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/pit.cpython-36.pyc create mode 100644 timm/models/__pycache__/pnasnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/registry.cpython-36.pyc create mode 100644 timm/models/__pycache__/regnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/res2net.cpython-36.pyc create mode 100644 timm/models/__pycache__/resnest.cpython-36.pyc create mode 100644 timm/models/__pycache__/resnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/resnetv2.cpython-36.pyc create mode 100644 timm/models/__pycache__/rexnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/selecsls.cpython-36.pyc create mode 100644 timm/models/__pycache__/senet.cpython-36.pyc create mode 100644 timm/models/__pycache__/sknet.cpython-36.pyc create mode 100644 timm/models/__pycache__/swin_transformer.cpython-36.pyc create mode 100644 timm/models/__pycache__/tnt.cpython-36.pyc create mode 100644 timm/models/__pycache__/tresnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/twins.cpython-36.pyc create mode 100644 timm/models/__pycache__/vgg.cpython-36.pyc create mode 100644 timm/models/__pycache__/visformer.cpython-36.pyc create mode 100644 timm/models/__pycache__/vision_transformer.cpython-36.pyc create mode 100644 timm/models/__pycache__/vision_transformer_hybrid.cpython-36.pyc create mode 100644 timm/models/__pycache__/vovnet.cpython-36.pyc create mode 100644 timm/models/__pycache__/xception.cpython-36.pyc create mode 100644 timm/models/__pycache__/xception_aligned.cpython-36.pyc create mode 100644 timm/models/__pycache__/xcit.cpython-36.pyc create mode 100644 timm/models/beit.py create mode 100644 timm/models/byoanet.py create mode 100644 timm/models/byobnet.py create mode 100644 timm/models/cait.py create mode 100644 timm/models/coat.py create mode 100644 timm/models/convit.py create mode 100644 timm/models/convmixer.py create mode 100644 timm/models/crossvit.py create mode 100644 timm/models/cspnet.py create mode 100644 timm/models/densenet.py create mode 100644 timm/models/dla.py create mode 100644 timm/models/dpn.py create mode 100644 timm/models/efficientnet.py create mode 100644 timm/models/efficientnet_blocks.py create mode 100644 timm/models/efficientnet_builder.py create mode 100644 timm/models/factory.py create mode 100644 timm/models/features.py create mode 100644 timm/models/fx_features.py create mode 100644 timm/models/ghostnet.py create mode 100644 timm/models/gluon_resnet.py create mode 100644 timm/models/gluon_xception.py create mode 100644 timm/models/hardcorenas.py create mode 100644 timm/models/helpers.py create mode 100644 timm/models/hrnet.py create mode 100644 timm/models/hub.py create mode 100644 timm/models/inception_resnet_v2.py create mode 100644 timm/models/inception_v3.py create mode 100644 timm/models/inception_v4.py create mode 100644 timm/models/layers/__init__.py create mode 100644 timm/models/layers/__pycache__/__init__.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/activations.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/activations_jit.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/activations_me.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/adaptive_avgmax_pool.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/blur_pool.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/bottleneck_attn.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/cbam.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/classifier.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/cond_conv2d.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/config.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/conv2d_same.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/conv_bn_act.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/create_act.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/create_attn.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/create_conv2d.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/create_norm_act.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/drop.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/eca.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/evo_norm.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/gather_excite.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/global_context.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/halo_attn.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/helpers.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/inplace_abn.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/lambda_layer.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/linear.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/mixed_conv2d.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/mlp.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/non_local_attn.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/norm.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/norm_act.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/padding.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/patch_embed.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/pool2d_same.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/selective_kernel.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/separable_conv.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/space_to_depth.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/split_attn.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/split_batchnorm.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/squeeze_excite.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/std_conv.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/test_time_pool.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/trace_utils.cpython-36.pyc create mode 100644 timm/models/layers/__pycache__/weight_init.cpython-36.pyc create mode 100644 timm/models/layers/activations.py create mode 100644 timm/models/layers/activations_jit.py create mode 100644 timm/models/layers/activations_me.py create mode 100644 timm/models/layers/adaptive_avgmax_pool.py create mode 100644 timm/models/layers/attention_pool2d.py create mode 100644 timm/models/layers/blur_pool.py create mode 100644 timm/models/layers/bottleneck_attn.py create mode 100644 timm/models/layers/cbam.py create mode 100644 timm/models/layers/classifier.py create mode 100644 timm/models/layers/cond_conv2d.py create mode 100644 timm/models/layers/config.py create mode 100644 timm/models/layers/conv2d_same.py create mode 100644 timm/models/layers/conv_bn_act.py create mode 100644 timm/models/layers/create_act.py create mode 100644 timm/models/layers/create_attn.py create mode 100644 timm/models/layers/create_conv2d.py create mode 100644 timm/models/layers/create_norm_act.py create mode 100644 timm/models/layers/drop.py create mode 100644 timm/models/layers/eca.py create mode 100644 timm/models/layers/evo_norm.py create mode 100644 timm/models/layers/gather_excite.py create mode 100644 timm/models/layers/global_context.py create mode 100644 timm/models/layers/halo_attn.py create mode 100644 timm/models/layers/helpers.py create mode 100644 timm/models/layers/inplace_abn.py create mode 100644 timm/models/layers/lambda_layer.py create mode 100644 timm/models/layers/linear.py create mode 100644 timm/models/layers/median_pool.py create mode 100644 timm/models/layers/mixed_conv2d.py create mode 100644 timm/models/layers/mlp.py create mode 100644 timm/models/layers/non_local_attn.py create mode 100644 timm/models/layers/norm.py create mode 100644 timm/models/layers/norm_act.py create mode 100644 timm/models/layers/padding.py create mode 100644 timm/models/layers/patch_embed.py create mode 100644 timm/models/layers/pool2d_same.py create mode 100644 timm/models/layers/selective_kernel.py create mode 100644 timm/models/layers/separable_conv.py create mode 100644 timm/models/layers/space_to_depth.py create mode 100644 timm/models/layers/split_attn.py create mode 100644 timm/models/layers/split_batchnorm.py create mode 100644 timm/models/layers/squeeze_excite.py create mode 100644 timm/models/layers/std_conv.py create mode 100644 timm/models/layers/test_time_pool.py create mode 100644 timm/models/layers/trace_utils.py create mode 100644 timm/models/layers/weight_init.py create mode 100644 timm/models/levit.py create mode 100644 timm/models/mlp_mixer.py create mode 100644 timm/models/mobilenetv3.py create mode 100644 timm/models/nasnet.py create mode 100644 timm/models/nest.py create mode 100644 timm/models/nfnet.py create mode 100644 timm/models/pit.py create mode 100644 timm/models/pnasnet.py create mode 100644 timm/models/pruned/ecaresnet101d_pruned.txt create mode 100644 timm/models/pruned/ecaresnet50d_pruned.txt create mode 100644 timm/models/pruned/efficientnet_b1_pruned.txt create mode 100644 timm/models/pruned/efficientnet_b2_pruned.txt create mode 100644 timm/models/pruned/efficientnet_b3_pruned.txt create mode 100644 timm/models/registry.py create mode 100644 timm/models/regnet.py create mode 100644 timm/models/res2net.py create mode 100644 timm/models/resnest.py create mode 100644 timm/models/resnet.py create mode 100644 timm/models/resnetv2.py create mode 100644 timm/models/rexnet.py create mode 100644 timm/models/selecsls.py create mode 100644 timm/models/senet.py create mode 100644 timm/models/sknet.py create mode 100644 timm/models/swin_transformer.py create mode 100644 timm/models/tnt.py create mode 100644 timm/models/tresnet.py create mode 100644 timm/models/twins.py create mode 100644 timm/models/vgg.py create mode 100644 timm/models/visformer.py create mode 100644 timm/models/vision_transformer.py create mode 100644 timm/models/vision_transformer_hybrid.py create mode 100644 timm/models/vovnet.py create mode 100644 timm/models/xception.py create mode 100644 timm/models/xception_aligned.py create mode 100644 timm/models/xcit.py create mode 100644 timm/optim/__init__.py create mode 100644 timm/optim/adabelief.py create mode 100644 timm/optim/adafactor.py create mode 100644 timm/optim/adahessian.py create mode 100644 timm/optim/adamp.py create mode 100644 timm/optim/adamw.py create mode 100644 timm/optim/lamb.py create mode 100644 timm/optim/lars.py create mode 100644 timm/optim/lookahead.py create mode 100644 timm/optim/madgrad.py create mode 100644 timm/optim/nadam.py create mode 100644 timm/optim/nvnovograd.py create mode 100644 timm/optim/optim_factory.py create mode 100644 timm/optim/radam.py create mode 100644 timm/optim/rmsprop_tf.py create mode 100644 timm/optim/sgdp.py create mode 100644 timm/scheduler/__init__.py create mode 100644 timm/scheduler/cosine_lr.py create mode 100644 timm/scheduler/multistep_lr.py create mode 100644 timm/scheduler/plateau_lr.py create mode 100644 timm/scheduler/poly_lr.py create mode 100644 timm/scheduler/scheduler.py create mode 100644 timm/scheduler/scheduler_factory.py create mode 100644 timm/scheduler/step_lr.py create mode 100644 timm/scheduler/tanh_lr.py create mode 100644 timm/utils/__init__.py create mode 100644 timm/utils/__pycache__/__init__.cpython-36.pyc create mode 100644 timm/utils/__pycache__/agc.cpython-36.pyc create mode 100644 timm/utils/__pycache__/checkpoint_saver.cpython-36.pyc create mode 100644 timm/utils/__pycache__/clip_grad.cpython-36.pyc create mode 100644 timm/utils/__pycache__/cuda.cpython-36.pyc create mode 100644 timm/utils/__pycache__/distributed.cpython-36.pyc create mode 100644 timm/utils/__pycache__/jit.cpython-36.pyc create mode 100644 timm/utils/__pycache__/log.cpython-36.pyc create mode 100644 timm/utils/__pycache__/metrics.cpython-36.pyc create mode 100644 timm/utils/__pycache__/misc.cpython-36.pyc create mode 100644 timm/utils/__pycache__/model.cpython-36.pyc create mode 100644 timm/utils/__pycache__/model_ema.cpython-36.pyc create mode 100644 timm/utils/__pycache__/random.cpython-36.pyc create mode 100644 timm/utils/__pycache__/summary.cpython-36.pyc create mode 100644 timm/utils/agc.py create mode 100644 timm/utils/checkpoint_saver.py create mode 100644 timm/utils/clip_grad.py create mode 100644 timm/utils/cuda.py create mode 100644 timm/utils/distributed.py create mode 100644 timm/utils/jit.py create mode 100644 timm/utils/log.py create mode 100644 timm/utils/metrics.py create mode 100644 timm/utils/misc.py create mode 100644 timm/utils/model.py create mode 100644 timm/utils/model_ema.py create mode 100644 timm/utils/random.py create mode 100644 timm/utils/summary.py create mode 100644 timm/version.py create mode 100644 train.py create mode 100644 util/__init__.py create mode 100644 util/__pycache__/__init__.cpython-36.pyc create mode 100644 util/__pycache__/html.cpython-36.pyc create mode 100644 util/__pycache__/util.cpython-36.pyc create mode 100644 util/__pycache__/visualizer.cpython-36.pyc create mode 100644 util/get_data.py create mode 100644 util/html.py create mode 100644 util/image_pool.py create mode 100644 util/util.py create mode 100644 util/visualizer.py diff --git a/README.md b/README.md new file mode 100644 index 0000000..4e8168c --- /dev/null +++ b/README.md @@ -0,0 +1,142 @@ +# ROMA +This repository is the official Pytorch implementation for ACM MM'22 paper +"ROMA: Cross-Domain Region Similarity Matching for Unpaired Nighttime Infrared to Daytime Visible Video Translation".[[Arxiv]](https://arxiv.org/abs/2204.12367) + +**Examples of Object Detection:** + + + + + +**Examples of Video Fusion** + + + +More experimental results can be obtained by contacting us. + +# Introduction + +## Method + + +- The domain gaps between unpaired nighttime infrared and daytime visible videos are even huger than paired ones that captured at the same time, establishing an effective translation mapping will greatly contribute to various fields. +- Our proposed cross-similarity, which are calculated across domains, could make the generative process focus on learning the content of structural correspondence between real and synthesized frames, getting rid of the negative effects of different styles. + + + +## Training +The following is the required structure of dataset. For the video mode, the input of a single data is the result of concatenating **two adjacent frames**; for the image mode, the input of a single data is **a single image**. +``` +Video/Image mode: + trainA: \Path\of\trainA + trainB: \Path\of\trainB + +``` +Concrete examples of the training and testing are shown in the script files `./scripts/train.sh` and `./scripts/test.sh`, respectively. + + + + +## InfraredCity and InfraredCity-Lite Dataset + + +
| InfraredCity | +Total Frame | +||||
|---|---|---|---|---|---|
| Nighttime Infrared | +201,856 | +||||
| Nighttime Visible | +178,698 | +||||
| Daytime Visible | +199,430 | +||||
| + | |||||
| InfraredCity-Lite | +Infrared Train |
+ Infrared Test |
+ Visible Train |
+ Total | +|
| City | +clearday | +5,538 | +1,000 | +5360 | +15,180 | +
| overcast | +2,282 | +1,000 | +|||
| Highway | +clearday | +4,412 | +1,000 | +6,463 | +15,853 | +
| overcast | +2,978 | +1,000 | +|||
| Monitor | +5,612 | +500 | +4,194 | +10,306 | +|
s-m89ldAYv)msj^XS2gXwwYgsf<@b@2Um+7(M-#d*hI&Wmr_nL_
zY3*0`?T(GIIjoMHj-#u7wf%ahj#_KDG-`Ajqvg)Bu6?43im;z(!Vc{HX3#jbI_E@H
zI8U_BiZDbiaQ4rOde0I|qVYuUtkQc=7t5lFw+q~A^bB!Mtl+&>w4N8MczaD;5G~BJ
zR@A;G)=;}H*2M;DH^fD;iRVReNnFNrQ+!EW!Sj+ZzN@vb9{*x@5GL*@7L#G%hNJN?
z7zI(9xPIjNdr2z&ZtCv&N#J&eev-J?KKSrX+x^;ysQJiGk*8mCd$DwddP%nH^?Rx7
z50lspqwa7bg2e5{qfs2Wm@7)qcjTubUVD?MO9jjsZ1=af+ $K3!rTs}%(TLxb3kSi|$m0f=eg=4LRlu`p@Lc;&z}@l2g*Z5w7AAe;1Fh
z6`%>^pF$GJT85mG3YP1gxNrTHY%*|c|Adu`TXcl~gkJ`x82?xDMf)-ly$GzkK%E_5
zF2YOxOJxxjsqmr?5?`}e4g+_XNXQD1Gn4?9={R6$x?s`91DFhcI~O6~Rv_hnL!k>y
zH3YI1F9h5TIlLWbhM0)mP&am9bJ!|;X5DT_g=?KTNvM*kXM~@
Sl#DV5qr|P;vF1H!H<23pLk-bH2z11l>^wU^+z6}>|+FZ
zDfLiJ+Ggb
z<
zr0MfDVAXv>RzrtGnaGkh=4`IEaym^C)RiYZ3^mLInn)d9WtL8~c25-Yr|t(hp4gqU
z4rBn|Cwbl++Zq$3)y>9=wc7|WBgmSP(M-IY*~{01CnFoWnOykk;;iuV9sY?Y6e(
ze4NlP*
tFW9(@g$;f_Efz-?g%c0sf!OVn=j=3CeihV>5bQj$xQWn8%IhNy0bBn>qr`iII78gV?~IOU;M@
z{Ca03ze_12U&T?Ah>R9)-kz4gSm$1nvc7|LCe>3S@
zv>UD!()$TztvEI9%}zd0qB+gtQ^l(}#1Q%|W$>|X!C2IGwC(i606ujkQ`nzE^H(l8
zlCS{LJboZ~`2~C^e;e*|>_A4xhwS?_KWE45q`A%{0XT!Djq9P}>Y5(W3=S+nwnr-c
zEG$~gLO&SR-^-R*cD&2{Q8h17{y`+WnZ@EbAJF~flmUz7L*)gzC(?-8_cVfMTPs=a
z=(Jlr=BIvlm%Kus
l>O
zlD3N^zcBa~EO*$H-UfK^UqTd<+Ixk!_j--FG3qaRS~-LD>lwUD1rEkOKPbXMlOqq0
zVIYCi{5nbH`2wPP8$Mfv!<8Cp`4@LR-E~EUi^LO}7DT10H&G`D_1wv+J=o>Sc+w
z*VVrQ8BDxd40NLfq^E$j&$a294-~WpNY^#b#u0zd3=ZJr8U>dJJnYXxC8;i_$%
z@8`LLX`&_q(Z7zANGaWQq4?w*8G5S7fH`aPW5$77Sy$S
zFF3$RSAUt~Ce)~IIk4HeI`XgC466te59@