spikezoo 0.2.2__py3-none-any.whl → 0.2.3.2__py3-none-any.whl

{spikezoo-0.2.2.dist-info → spikezoo-0.2.3.2.dist-info}/RECORD

@@ -1,8 +1,6 @@
-spikezoo/__init__.py,sha256=edeJMa3xcfGEZTxMO4Z7YZTh4gtBnYkCGrT7aOYse2c,283
+spikezoo/__init__.py,sha256=3z97Jy20aDJoK3e1ECXneY-i5jLj9Idop5ClHeJLYGE,604
 spikezoo/archs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-spikezoo/archs/__pycache__/__init__.cpython-39.pyc,sha256=Ba5cgbZ2jiTexmzMbqpADlFBQGzqLPtU-YbGhY9wJa0,175
 spikezoo/archs/base/nets.py,sha256=j-XDEkntYl4M6xe-jqyUOhFBlhh8efNwy4qu-WVqjc4,1311
-spikezoo/archs/base/__pycache__/nets.cpython-39.pyc,sha256=Ie9588wEf8hRFO7donRgviMdoSK4jeeGZn56FDa_tg8,1554
 spikezoo/archs/bsf/README.md,sha256=maT9K0dZcwFPiYWmFISVogF-INTwNr6alqHSNKlk7G0,2777
 spikezoo/archs/bsf/main.py,sha256=at4CWWqaoGo1k6PqRBOi_PixtDPmND6_7mU6LvWWnLI,15136
 spikezoo/archs/bsf/requirements.txt,sha256=ZQCaKDZAJvmFtKPcbDSIP3gBpg_YdvwCpaDkS__kfHE,79
@@ -14,12 +12,9 @@ spikezoo/archs/bsf/metrics/psnr.py,sha256=OntyhZtYIKEbdy5w-qwkl6mBt767W5pitDEjMm
 spikezoo/archs/bsf/metrics/ssim.py,sha256=RxVoEMJPgu370DWfDRE01UnTOorh-Xy0DldXQFhAi4o,1818
 spikezoo/archs/bsf/models/get_model.py,sha256=acp0f10J81ue_lSJHlMfGeMq7IijZmPsw9Trg0nqKiQ,129
 spikezoo/archs/bsf/models/bsf/align.py,sha256=X_Ud0oCZSYGFQ8DWvOG4yozUaDOJi4X44vBiXsBThiI,7979
-spikezoo/archs/bsf/models/bsf/bsf.py,sha256=cgm_nEl-XMWUz7EOm8aVmfqB5PorT1YsP9GZ-57rNMY,3480
+spikezoo/archs/bsf/models/bsf/bsf.py,sha256=W3xwHXcKODJqfSRc_Kn-7C_YjVGpse_mZ2tbrDJ6w0Q,4060
 spikezoo/archs/bsf/models/bsf/dsft_convert.py,sha256=xpFwWFl1ms9LxaA96xdDOf-h_S6foScc3oh-nGjSG-o,3110
-spikezoo/archs/bsf/models/bsf/rep.py,sha256=xD0nNjEQdmBwzMlUEQxYmLmZBx8ieGwbIy-v_1CeXw8,1643
-spikezoo/archs/bsf/models/bsf/__pycache__/align.cpython-39.pyc,sha256=vGZzrpGGFZeVzf-FGdA3Sq8u87y7CZ-wC93vgPwkT4E,7082
-spikezoo/archs/bsf/models/bsf/__pycache__/bsf.cpython-39.pyc,sha256=XreuEfY7R00NN4GXjZQpt_x44n6vIfjDher7qNOWq5c,4216
-spikezoo/archs/bsf/models/bsf/__pycache__/rep.cpython-39.pyc,sha256=RtKNwwHV0-vQ3AKGyDUeaeDVFopQLsVDEIdqklJIAxw,1860
+spikezoo/archs/bsf/models/bsf/rep.py,sha256=Y3YPADL6ndu4u7RwYUFqmGVUqzW0HbgXKu4Z7x52Alg,1660
 spikezoo/archs/bsf/prepare_data/DSFT.py,sha256=RDFREQc-pAGxpETgb1umjQNnLAmpqsRpEmsrjrG4hEU,2203
 spikezoo/archs/bsf/prepare_data/crop_dataset_train.py,sha256=CpKIhI8kc5TzWMGSHY33IlROBTrXrY0kVGxKLvwjcvo,6050
 spikezoo/archs/bsf/prepare_data/crop_dataset_val.py,sha256=Zrfe2rsnHXB-TikO9J6s0SL_u0jg492NgXSCebgvK5A,6009
@@ -27,39 +22,16 @@ spikezoo/archs/bsf/prepare_data/crop_train.sh,sha256=VoIqvQ1TWSj5uvkcp6EZIo28egI
 spikezoo/archs/bsf/prepare_data/crop_val.sh,sha256=PY45EGOvdn89hAUWmeIwHrzzMWmwBbqGYgh0y5oYzP4,185
 spikezoo/archs/bsf/prepare_data/io_utils.py,sha256=GUs7ocNekOKMSfMNjHAWbZFSOWXiLRtboQLl5NiY-CI,1850
 spikezoo/archs/spikeclip/nets.py,sha256=j2rPD3AFWLl142XxmQL4PxWw06f4gHi8zjBUjndJ8pQ,1433
-spikezoo/archs/spikeclip/__pycache__/nets.cpython-39.pyc,sha256=Bp7ISUzHzcIj5ebzUzry2pDrWdg1dlr-GH72wWTHkiM,1570
-spikezoo/archs/spikeformer/LICENSE,sha256=YYnrgInlCj2xhRzxoWJZDx6DGHe85AxmozL-SyCrW8k,1069
-spikezoo/archs/spikeformer/README.md,sha256=e1OWqkquJpkAG-TpzB6M0Eaob2-I0zgYgk70kanLoOk,1089
-spikezoo/archs/spikeformer/evaluate.py,sha256=4orJqfDyfXNiRJpOJP_JdybdcoGHMV8ugMMhGn6WX8Q,2986
-spikezoo/archs/spikeformer/recon_real_data.py,sha256=GxWPYJLJRpzAp0p1XfH1wBlgt5aY-RH6UxVRm7AV17s,3250
-spikezoo/archs/spikeformer/requirements.yml,sha256=v-9a7_nyM1Rj6rZPiXPD9pxMIm0C1BO9qOyjCgdHQzI,2856
-spikezoo/archs/spikeformer/train.py,sha256=t8N38QUw4NagABV5gkgWi_UssbDQOcTmOgGmZJDrToA,6655
-spikezoo/archs/spikeformer/utils.py,sha256=DK0qJA01_Hs0VL9pcdyHdctJoJVr2zTPkWhk-6aT6yk,821
-spikezoo/archs/spikeformer/CheckPoints/readme,sha256=h_HwP9-B0O3-fBTUVNNXcoGzVV6KBFakWSlXJGlpIRg,47
-spikezoo/archs/spikeformer/DataProcess/DataExtactor.py,sha256=N_OMPASjaqXOQhGokGPehrMUwaxgh9ybF_diVycRuA8,1319
-spikezoo/archs/spikeformer/DataProcess/DataLoader.py,sha256=28T5hlS1qZVhHbtn1CS22V7uqX_BXlD-GqAz3HyoFOc,2951
-spikezoo/archs/spikeformer/DataProcess/LoadSpike.py,sha256=sR4tIiKLBxEyF8bzNjLJsILmBNSlYjKs8On_-c-M3yk,1580
-spikezoo/archs/spikeformer/EvalResults/readme,sha256=o3kRLSB6ZZPRW4naL9EghJuMvsLOnrHKauVMfWItnBM,83
-spikezoo/archs/spikeformer/Metrics/Metrics.py,sha256=YxfJSL_8UqASCy2R2G-ZCumdRFGNFCxksk8PlMwpStM,1240
-spikezoo/archs/spikeformer/Metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-spikezoo/archs/spikeformer/Model/Loss.py,sha256=evR1RuiU-A63p1llK_TdFJ0BCyf9l2yVraoKzzlMzEM,3183
-spikezoo/archs/spikeformer/Model/SpikeFormer.py,sha256=W_nSR4m2YlE_EUaVNcz_eB8QO3Ns5dJUhOzIPL20yZc,7171
-spikezoo/archs/spikeformer/Model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-spikezoo/archs/spikeformer/Model/__pycache__/SpikeFormer.cpython-39.pyc,sha256=_ofZvPDudojhpBGitJ9BP-lHQPlVLH5hwo1fi8W2E3E,7057
-spikezoo/archs/spikeformer/Model/__pycache__/__init__.cpython-39.pyc,sha256=QxKCbkXZoN8TlFT05KDorzQWZH4eXsOTYTl0y_-J-8o,193
 spikezoo/archs/spk2imgnet/.gitignore,sha256=LvYh4-uHW8ZL6P5S7I6f35ZaKULvuVoBQ06ia6r1llM,2148
 spikezoo/archs/spk2imgnet/DCNv2.py,sha256=KqAWzoOQFX1eEqaIP90Ahhj88qvk2K-J21WyzOMQwt4,4715
 spikezoo/archs/spk2imgnet/align_arch.py,sha256=NpEDZy4YX2JD6mNrw1FOfyPpMqQ866ylYktl-kRzwIU,6264
 spikezoo/archs/spk2imgnet/dataset.py,sha256=ROGBtAYtIfTLAmpCUpEkYeLWTAC8QGMpS8Tfq7jK-K8,4794
-spikezoo/archs/spk2imgnet/nets.py,sha256=_VFExW8fLcxrjAf8WYgqKgZhliOa5WI_lziVFOpa7nI,8039
+spikezoo/archs/spk2imgnet/nets.py,sha256=ELEqYElUWp-NNHA9cF2ABweWiSVZ9IJYb2OLQ106WgQ,8061
 spikezoo/archs/spk2imgnet/readme.md,sha256=0uYl7DEeXQ7EmioqIHAkmkWXIcvxpY6n12hFTwpyjr0,2591
 spikezoo/archs/spk2imgnet/test_gen_imgseq.py,sha256=UpqerSdtF5qMODIxAuVY3JnhsPy8GjNQA_4xl5X7dwk,4359
 spikezoo/archs/spk2imgnet/train.py,sha256=ncNwAhFlAhDd4rMkSeAjBhRECp2En1hAZn2-RH7jVpU,6195
 spikezoo/archs/spk2imgnet/utils.py,sha256=Gc-05AJDfiXqkDAaiTeLBa2oiD78l1PPrBP6frlFy30,1924
 spikezoo/archs/spk2imgnet/.github/workflows/pylint.yml,sha256=lNUdbM2y3yOGPPrGUNDjnQL3fJnswtFIsuLpchBfLAc,553
-spikezoo/archs/spk2imgnet/__pycache__/DCNv2.cpython-39.pyc,sha256=1mSOxJwgg-5bCAwqgIp6VOUWmHpaRv4ILnkgEiCGaI0,4095
-spikezoo/archs/spk2imgnet/__pycache__/align_arch.cpython-39.pyc,sha256=WqnFKRXqXZY74-4CBp9vpebg98tzwrSsvXibOU1AzO8,4361
-spikezoo/archs/spk2imgnet/__pycache__/nets.cpython-39.pyc,sha256=Kam188Xp8paFk0ZhgzHfOrNibFDExu53MQNge5aM6lI,5862
 spikezoo/archs/ssir/README.md,sha256=0e56N8SqYfDRgyoyxaMctJqJ2DeWdQ4NUlLDQEUr2O0,2518
 spikezoo/archs/ssir/losses.py,sha256=mUEghvJQoLM4bTs0AmDa5sZedEHWo_GksW2ojNWYYC8,623
 spikezoo/archs/ssir/main.py,sha256=Pn0mmUST4hg_BJPOU6NWvorjuZ0OHncFOzVEtxDEPQQ,12054
@@ -74,17 +46,13 @@ spikezoo/archs/ssir/metrics/psnr.py,sha256=OntyhZtYIKEbdy5w-qwkl6mBt767W5pitDEjM
 spikezoo/archs/ssir/metrics/ssim.py,sha256=RxVoEMJPgu370DWfDRE01UnTOorh-Xy0DldXQFhAi4o,1818
 spikezoo/archs/ssir/models/Vgg19.py,sha256=BKYf51YqQantkuxGM5S3yD2a5Pf2nYBzEfmA0XqTjGU,1435
 spikezoo/archs/ssir/models/layers.py,sha256=gYShN5cp3B1GaNmQD5_6CpYSt6k6h0cZ5IJNtrt0dCw,3450
-spikezoo/archs/ssir/models/networks.py,sha256=LtcYT_SU3-X-XFCDfsyMMbe1lPe0I0eiW8oNClwsjdE,1948
-spikezoo/archs/ssir/models/__pycache__/layers.cpython-39.pyc,sha256=jUAodYnILqoNJWWU2P5b1EgGItM2cSppoYWeQNKtals,3753
-spikezoo/archs/ssir/models/__pycache__/networks.cpython-39.pyc,sha256=XiOQdwmw-VLLBw4LEvE8PWP8nouKX6IXu_HXOB8DGHw,2799
+spikezoo/archs/ssir/models/networks.py,sha256=-qwwwC9SWcOzf_TcswnudoOVSjZNvRMiAg5-NRDl14I,1946
 spikezoo/archs/ssir/shells/eval_SREDS.sh,sha256=byjDfNb_NAO8z28L7Laktlc3qYZjfeY5qn0pkMmql9E,112
 spikezoo/archs/ssir/shells/train_SSIR.sh,sha256=y-LlaWNqOKwGUXKm2NodCgM3LYOIans25kN7CgDJ9z4,256
 spikezoo/archs/ssml/cbam.py,sha256=hfVI1vYpboEPRBMKWqWjVlqX41XQi7A4Pwou5PJlPXo,8869
-spikezoo/archs/ssml/model.py,sha256=ILN9LRoY_WZYASMSSUsCpQPabAF2kybPkW4OwCiU_nE,10220
+spikezoo/archs/ssml/model.py,sha256=DqTwDbwS7diZPxjVz580lAarPeZZ43EtaNjMJWN2Ujo,10354
 spikezoo/archs/ssml/res.png,sha256=o8VLsy8-znCM9ZoSbsBmV3dTd8O0R48JWyNtuIekQIY,37233
 spikezoo/archs/ssml/test.py,sha256=3yrMAWDBdhpyVjqNeuDtz1s2XemrR9ZXZDo8yTUAfac,2036
-spikezoo/archs/ssml/__pycache__/cbam.cpython-39.pyc,sha256=wtXvwDBTBBEeRhfmXeNAmbUN6puFxomELtQHk-xHnxA,8678
-spikezoo/archs/ssml/__pycache__/model.cpython-39.pyc,sha256=tgYpLrpWboUAruKwXbzvdCe78ODXryO0lGKgRt2u7to,10192
 spikezoo/archs/stir/.git-credentials,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spikezoo/archs/stir/README.md,sha256=WA0QvTGiH8kDQ4mNwj4lnZHypKSDVf_UWXckXnXltJc,3223
 spikezoo/archs/stir/eval_SREDS.sh,sha256=dihTfrrr0jbkXEbpEStKgsk-t5L_4ehu9mpeMliJbME,144
@@ -97,16 +65,13 @@ spikezoo/archs/stir/configs/utils.py,sha256=ebPww68XHeS2Ip8G8FD7AxRDh2-cPk1VFKBP
 spikezoo/archs/stir/configs/yml_parser.py,sha256=4IMJSoMKyBU00Qv4K7mmRBT0YFCLPHb7h7J0pIGtBLo,2078
 spikezoo/archs/stir/datasets/dataset_sreds.py,sha256=lN-woOl1RJ1Q5t7ttPIZizdkUMtYCTfXj_LAn78kqr4,6803
 spikezoo/archs/stir/datasets/ds_utils.py,sha256=RfQyC_8Y50-R-xnxlGooNwAoaTvviIlSQrrRIgDMGqc,1939
-spikezoo/archs/stir/metrics/losses.py,sha256=BDMKLMrAb7L_4zWzS3nXoW9RKjFrYyl9PeiZpYoW5V4,7753
+spikezoo/archs/stir/metrics/losses.py,sha256=pOs0XYZuKPIjppWwEmx8CXpDPqhq5QcR0NMMUZUR01o,7768
 spikezoo/archs/stir/metrics/psnr.py,sha256=OntyhZtYIKEbdy5w-qwkl6mBt767W5pitDEjMmnqjRo,707
 spikezoo/archs/stir/metrics/ssim.py,sha256=RxVoEMJPgu370DWfDRE01UnTOorh-Xy0DldXQFhAi4o,1818
 spikezoo/archs/stir/models/Vgg19.py,sha256=BKYf51YqQantkuxGM5S3yD2a5Pf2nYBzEfmA0XqTjGU,1435
-spikezoo/archs/stir/models/networks_STIR.py,sha256=KA0_qYBDcD_DJ6eafmvYlSpRyfVt8i8rG88BQ-IBAI4,15290
+spikezoo/archs/stir/models/networks_STIR.py,sha256=dU19BT2sAZMa-avJPdQvC48orMFUYsE05ZzWPIZA9Sg,15746
 spikezoo/archs/stir/models/submodules.py,sha256=gr0W8_ghP6pF5E5M1Ii58XYXOzR5ox8n0Xoh0vDAv6c,3360
 spikezoo/archs/stir/models/transformer_new.py,sha256=INZFO156bD4A0t5agChPT87uPDJXiu9gibXMORZgzxk,6343
-spikezoo/archs/stir/models/__pycache__/networks_STIR.cpython-39.pyc,sha256=GZ5MB9hqinl17F51CtdfYw0EoC75eQF4SGhrwb_WcKk,13282
-spikezoo/archs/stir/models/__pycache__/submodules.cpython-39.pyc,sha256=zVNI7hYN7S2yvTKYJt8FIXSU5yUg1rfE4tyZ5zPrsWo,3640
-spikezoo/archs/stir/models/__pycache__/transformer_new.cpython-39.pyc,sha256=N1ZPGfGsHXI9b1bwRAOh40eJWlmYfmK7DF9qDr248d0,5939
 spikezoo/archs/stir/package_core/setup.py,sha256=l0ZAYjzpqI6IvNFm5pHOmf5jmapFXqJSdOk8SBNupc0,112
 spikezoo/archs/stir/package_core/build/lib/package_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spikezoo/archs/stir/package_core/build/lib/package_core/convertions.py,sha256=aKnq_wJ8sddEI2T0ITxukpDdiKsdt-zf3nzM2CHFxGs,26159
@@ -139,28 +104,22 @@ spikezoo/archs/stir/package_core/package_core/net_basics.py,sha256=QX_5zlC5-_ITT
 spikezoo/archs/stir/package_core/package_core/resnet.py,sha256=l93GwsKVBl75tUMYLZWkzZjNhO3B_Yoissb3oUdq3zE,13627
 spikezoo/archs/stir/package_core/package_core/transforms.py,sha256=_hE5Y6EWsxacwcfdI1jS-wCvwGkA32-k-4XLPVhquDY,3779
 spikezoo/archs/stir/package_core/package_core/utils.py,sha256=icSibxXKqEZyHL8GU1J0PMahCfxwVSwBtLGW_kHV25g,2316
-spikezoo/archs/stir/package_core/package_core/__pycache__/__init__.cpython-39.pyc,sha256=kMkbhE8EH_BDTqvZladciovDa11NWsKJTLZor2XXliE,206
-spikezoo/archs/stir/package_core/package_core/__pycache__/net_basics.cpython-39.pyc,sha256=ZHXtxnR4tYqkTl79drTDby1cu2c_SzgYBLQMMuyVwYQ,3622
 spikezoo/archs/stir/package_core/package_core.egg-info/PKG-INFO,sha256=2njov-JTXZp2Sgwyx7KSL0fIPNyOR-lUYSSxIiDQH_Q,56
 spikezoo/archs/stir/package_core/package_core.egg-info/SOURCES.txt,sha256=rTCDnAkAo4JuMaJgXeumnUODeQBpefgwT-dqrIUXoRc,541
 spikezoo/archs/stir/package_core/package_core.egg-info/dependency_links.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 spikezoo/archs/stir/package_core/package_core.egg-info/top_level.txt,sha256=ezjGZVvZhOw8f-HRDngFtMvGh0NfvyT3sKcG4sSOSoc,13
-spikezoo/archs/tfi/nets.py,sha256=Sw4Oy92DS2DqrxzS7P02-cN1JQDq35SnngcbrM8VTEk,1377
-spikezoo/archs/tfi/__pycache__/nets.cpython-39.pyc,sha256=4fzCYhy51OyXFeAtEWzNCXCRPDtsRDyS3kfGByg5zhg,1393
-spikezoo/archs/tfp/nets.py,sha256=e1UzuOizrXX3AaW0fvIvvjyqZ905La2EFCWJlytk7A0,383
-spikezoo/archs/tfp/__pycache__/nets.cpython-39.pyc,sha256=HsOsttvdfAoTFsq1HTsFhaDnOU9ZJ6LFZG0XSgvNMYk,847
+spikezoo/archs/tfi/nets.py,sha256=IpXGoemHjan6FpFZjt2VU-pWE6AptsTlCFf20ha86zo,1382
+spikezoo/archs/tfp/nets.py,sha256=mNngiPBEXcNH4yP6PiwOgsTS8dOhHvdnXq-UNuhfpxY,388
 spikezoo/archs/wgse/README.md,sha256=vUKBdCOV1MMr3ZqfXgiim99dYTERinrkzejhY-uwoiQ,3151
 spikezoo/archs/wgse/dataset.py,sha256=pCvOrFRHn7tCku1bAi9vLL_tPIZQnwj57mfvSjnwFgc,1822
 spikezoo/archs/wgse/demo.png,sha256=6SdZmRf6WYd6OHa1ll0F8msbnR_gsHiuqR3OLmYi1fU,64157
 spikezoo/archs/wgse/demo.py,sha256=9lQDIi5at95ky0ccRdcdYYxiA9U8anX2DsXTlZTIQaQ,2416
-spikezoo/archs/wgse/dwtnets.py,sha256=fyLM6O1GsPVGvaZwzq0kz8jc9kVS5bElpFbLTjbrfKY,4089
+spikezoo/archs/wgse/dwtnets.py,sha256=oCKQspqO_61IkXdFMxKTlK6HXge2FcqfObx9Llb426U,4096
 spikezoo/archs/wgse/eval.py,sha256=5Daj-1eoYRqjgp3hf6W3m54jtcVoydBPBb0bR0ZIDF0,3888
 spikezoo/archs/wgse/submodules.py,sha256=qFsOnAFx7uwvIo9ymUPm3Yo6JvYnZYhrnnJyqFoPJWw,2326
 spikezoo/archs/wgse/train.py,sha256=8y8rjTuTFiSnYR6wWibk_mTszsINV995BoO8nxR_u18,9361
 spikezoo/archs/wgse/transform.py,sha256=bX3jPacCJdOo1FZmDgIZgS5DWrkUs3kw8njJeHh0NLQ,4532
 spikezoo/archs/wgse/utils.py,sha256=UXTo8HoeB4BwSLXSbi3AyM1tokPnJ--Giz9ln2Yr0nQ,3892
-spikezoo/archs/wgse/__pycache__/dwtnets.cpython-39.pyc,sha256=U0zs8m7ZzDRqrclKEeU5z5YFBBHjEbzChrOCXWMMp-I,3780
-spikezoo/archs/wgse/__pycache__/submodules.cpython-39.pyc,sha256=J5JCZwJBhWRP1JQXCrxPes1tJBg_ggKVym81aTsBpYQ,2179
 spikezoo/archs/wgse/logs/WGSE-Dwt1dNet-db8-5-ks3/log.txt,sha256=99XvHRXAhKc8E6JwP8fFBBjjFFEvrtOL2y17ibYXiZc,990
 spikezoo/archs/wgse/weights/demo.png,sha256=jy3xM3Fe_A4b79wbxHoiPGHzMmfK_LMcMs3Y9nT1i3o,69728
 spikezoo/data/base/test/gt/200_part1_key_id151.png,sha256=hkKTqpvv1Ms_xjcP3lQ2pyTswiCM1I7YLKR_ANqD5Bk,52637
@@ -175,37 +134,41 @@ spikezoo/data/base/train/gt/203_part4_key_id151.png,sha256=xUfdlXNWdPlRshLOaEF6u
 spikezoo/data/base/train/spike/203_part2_key_id151.dat,sha256=YEenLmbPvcxnKkVn3O7yDVYb-UwpM5OPlRhxVWLYy3Q,3762500
 spikezoo/data/base/train/spike/203_part3_key_id151.dat,sha256=MY9nM6XzKj-P-tRQ33WZ3G5xulNTpAXKP0y8ZQo7AIQ,3762500
 spikezoo/data/base/train/spike/203_part4_key_id151.dat,sha256=IVi2jics66YzpIF-WTkw47te4qOj9cjdgz56GmHpJKg,3762500
-spikezoo/datasets/__init__.py,sha256=bnskj7Bo0cWEk8XEBjOB2etTOjtsoReOxjezeZmYyTs,3274
-spikezoo/datasets/base_dataset.py,sha256=gm3jLmnkbiWoWZuyjnGySU2WVgkv-tdMWLRuG5lGYvI,5787
-spikezoo/datasets/realworld_dataset.py,sha256=iqn9oYu56Ph2nsA2U5aesTjflNmfr3-6krHVXC49ByM,715
-spikezoo/datasets/reds_small_dataset.py,sha256=nhX95iWAADlu65XCGsU8CosjLHvxzSgAumA4MDwaTUc,883
-spikezoo/datasets/szdata_dataset.py,sha256=2zLLKj6PC2XUcmgUMUqKKiA_zQZdZx7hCsI8Ft-i4o8,1230
-spikezoo/datasets/uhsr_dataset.py,sha256=cuQBEuOw0AFaxV-reLWqsS2KevKTYF2Pr8ak5dSlkbc,1166
+spikezoo/datasets/__init__.py,sha256=lRJsvCfgbe3qrd9BKTlG9dsgfIJbfXqWOynnlAcBiUI,3346
+spikezoo/datasets/base_dataset.py,sha256=gaMcrgWB3JLfc5lAM5wZjgMj72lZaO5otGfXOwRRiL8,5954
+spikezoo/datasets/realworld_dataset.py,sha256=CSyXj_uo0y1a4TlLDSTkB0CyFiFdZiU1phrWdkwvqgg,701
+spikezoo/datasets/reds_base_dataset.py,sha256=W-IJv9H1bsKgp3RT3zsV40jw2PqY2M76jtIS4Qpif1o,859
+spikezoo/datasets/reds_ssir_dataset.py,sha256=t0hm6PUWX5hfvSXB0UEv_JuihIhc5-mufrfur5bIq_0,7076
+spikezoo/datasets/szdata_dataset.py,sha256=7lSqNzWkVUVTVufPhm7AjpLghpZvjfu1X9c0sCsrjwM,850
+spikezoo/datasets/uhsr_dataset.py,sha256=Ue0vfKMpmvR0TelQx8G4xdWB8HdSlb4HSjIVOyFe5oE,1168
 spikezoo/metrics/__init__.py,sha256=LIKeWNeEMZLANITQD68XJBOhDq7iHiKC7ExtdrXMyQs,3273
-spikezoo/models/__init__.py,sha256=ePC94bz7Yjvg_CQQSWCkb7J5E_Tc370408a9tkzDFDQ,1679
-spikezoo/models/base_model.py,sha256=BQmiJSRHfuzeCDQOc1eoAaso3E0SWqKAtwev7-cSF18,6695
-spikezoo/models/bsf_model.py,sha256=KVKmukMfhr8hOIwIlWAKnCvDSHlXK2HJwjzP9bEGpNw,3523
-spikezoo/models/spikeclip_model.py,sha256=ltEbEb-TNyzHGWHgiOc6-701LSGvSHYiD1WXglwgOJM,1010
-spikezoo/models/spikeformer_model.py,sha256=oNwmzWNdemJhjDWnPoJwlxUzKLAAu9YpE2J9iwEkEoA,1586
-spikezoo/models/spk2imgnet_model.py,sha256=QknQeqLfZdU2MoToUsNqFvAGN0rN2-6wzOw6I4Xtqwc,1549
-spikezoo/models/ssir_model.py,sha256=sVLZ_7BwBqloB3H94BuQfyhbepwS1FdoEtbIkrPqQkk,586
-spikezoo/models/ssml_model.py,sha256=u_UohCL2Q_iRZ-I_udJiQ5_30_ZASKxhttDgJKLtZ6E,493
-spikezoo/models/stir_model.py,sha256=fXFajxWs2P4OoAPGXGkwOGJCDslWqJ4wus4mfbvvH7w,1176
-spikezoo/models/tfi_model.py,sha256=fLaBOz1f3c-wvA8bEYVfUNk9vgtbMhbS02-Xie294mg,536
-spikezoo/models/tfp_model.py,sha256=khG15qA_32PEb3BaYHFo5BzXUaxh-Napl_0bwXIcGz8,535
-spikezoo/models/wgse_model.py,sha256=9N1O_ucbdQ_lndLpWNjuhidyWKs8Ct8Wr-OWTYCVc44,860
+spikezoo/models/__init__.py,sha256=QZTELBoM3bUW8jZoxN4OuA2RYKeVUT1fboyeIuK8Rtk,1722
+spikezoo/models/base_model.py,sha256=v3TD4AmjttTZUg0vEy736TOFdbbBgDLZg_RL-b4-vYM,9152
+spikezoo/models/bsf_model.py,sha256=XeZcVC_ODJxyS_I6-CtzlHXSWntgsUtbuAKjczIQ_0M,3972
+spikezoo/models/spcsnet_model.py,sha256=kLzv-ASXZGnqEFx0jUBONBeRCrsnQ_omkQUYEnr6uJc,540
+spikezoo/models/spikeclip_model.py,sha256=Ej84RuYbkFRthtBMV1JtmTkUshAqINlrrJ7yiKIsC9s,1125
+spikezoo/models/spk2imgnet_model.py,sha256=ghdO1oECrRBijvGNT89H5XDi9CvWfbY3wOzmuCrgMJ0,1565
+spikezoo/models/ssir_model.py,sha256=8tg36eLKAQEOfYiFW-XyO6RfvCLxUXz0fVhTtZ-Dw-Q,625
+spikezoo/models/ssml_model.py,sha256=pCo2Wp38cRWSqGFEddsteEby_My4Rp-MKIx_g4kjoHo,2380
+spikezoo/models/stir_model.py,sha256=htsqhOboIc3GWvQc9gXxyaSI2SYR3TdZR691uV5LZ5s,2193
+spikezoo/models/tfi_model.py,sha256=tgD_HsiXk9jGuh5f_Bh6c3BqJi1p5DWCVo4N1tp5fgs,663
+spikezoo/models/tfp_model.py,sha256=ihl1H__bWIbE9oair_t8rNJ5qnPJPKl-r_DpaO-0Sdk,663
+spikezoo/models/wgse_model.py,sha256=Kl9uV-LeO0Lj7SuPQ9pglw1Khs2b-7miS3A_faL6WSU,805
 spikezoo/pipeline/__init__.py,sha256=WPsukNR4cannwsghiukqNsWbWGH5DVPapR_Ly-WOU4Q,188
-spikezoo/pipeline/base_pipeline.py,sha256=n6EXiNDLeEdej1fuDfPbrbGxwCseVyJeR_Tq0ipDmBo,12265
-spikezoo/pipeline/ensemble_pipeline.py,sha256=Aoi0lLcSDi9aJGIyHsjs45OYBqjDBDu97353IHLoUmw,2467
-spikezoo/pipeline/train_pipeline.py,sha256=xMOX-ulb1fPLOH7EIK4ww8q1SY8x_lJvnGov_kXZl-k,4060
+spikezoo/pipeline/base_pipeline.py,sha256=ns2rBRbKaq8M9yDtWOiD-t6lo_5_c5rE2ZWH9sR78P4,13361
+spikezoo/pipeline/ensemble_pipeline.py,sha256=ljZkGiCCpxvpC04Aa-r_tvBnqcBpUVi9fl_878tJAcg,2555
+spikezoo/pipeline/train_cfgs.py,sha256=6NO7DfPc7yjJfOrcIPQPfUPbUODz6eRKurEIDjMmaxA,3836
+spikezoo/pipeline/train_pipeline.py,sha256=BgHUsdv33B_OKauOVclNt7yIPb-_O-93ZHLHIjrwWaA,8459
 spikezoo/utils/__init__.py,sha256=bYLlusAXwLCoY4s6nhVgviax9ioRA9aea8qgRmj2HpI,152
 spikezoo/utils/data_utils.py,sha256=mk1xeyIb7o_E1J7Z6-gtPq-rpKiMTxAWSTcvvPvVku8,2033
 spikezoo/utils/img_utils.py,sha256=0O9z58VzLxQEAuz-GGWCbpeHuHPOCpgBVjCBV9kf6sI,2257
-spikezoo/utils/other_utils.py,sha256=fKqs4zRxzQsIfmYZv02PZlVaGrmVEjq2KHTMrk_tBKY,2845
-spikezoo/utils/spike_utils.py,sha256=0GY1hQCOCj0HDDwjXxrHykdjTKmPdb9rC_CexpRzwdk,3123
+spikezoo/utils/optimizer_utils.py,sha256=jvcd4zTY2LCJH6wCwOZ0lsAuJQm6LIVzbprLO3ojYCY,744
+spikezoo/utils/other_utils.py,sha256=uWNWaII9Jv7fkWNfkAD9wD-4ID-GAzbR-gGYT-1FF_c,3360
+spikezoo/utils/scheduler_utils.py,sha256=5RBh-hl3-2y-IomxMs47T1p3JsbicZNYLza6q1uAKHo,828
+spikezoo/utils/spike_utils.py,sha256=u4Haa6Sp5xFqs61ztvq161oXTA_aZmNW3VYUZcayNW0,4296
 spikezoo/utils/vidar_loader.cpython-39-x86_64-linux-gnu.so,sha256=uXqu7ME---cZRRU5LUcLiNrjjtlOjxNwWHyTIQ10BGg,199088
-spikezoo-0.2.2.dist-info/LICENSE.txt,sha256=ukEi8E0PKq1dQGTXHUflg3rppLymwAhr7il9x-0nPgg,1062
-spikezoo-0.2.2.dist-info/METADATA,sha256=j-XErZpa-tDx5wkFwwbOaHdcUjRuuGopldedu0hwEVk,7939
-spikezoo-0.2.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-spikezoo-0.2.2.dist-info/top_level.txt,sha256=xF2iuOstrACJh43NW4dsTwIdgKfXPXAb_Xzl3M1ricM,9
-spikezoo-0.2.2.dist-info/RECORD,,
+spikezoo-0.2.3.2.dist-info/LICENSE.txt,sha256=ukEi8E0PKq1dQGTXHUflg3rppLymwAhr7il9x-0nPgg,1062
+spikezoo-0.2.3.2.dist-info/METADATA,sha256=k4-w6mwZAfSx1YhunWGVx3LYuocZEqs8ypcMr9WhkZ8,11962
+spikezoo-0.2.3.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+spikezoo-0.2.3.2.dist-info/top_level.txt,sha256=xF2iuOstrACJh43NW4dsTwIdgKfXPXAb_Xzl3M1ricM,9
+spikezoo-0.2.3.2.dist-info/RECORD,,

spikezoo/archs/spikeformer/CheckPoints/readme

@@ -1 +0,0 @@
-This is a folder for saving the trained model !

spikezoo/archs/spikeformer/DataProcess/DataExtactor.py

@@ -1,60 +0,0 @@
-import os
-import numpy as np
-
-
-class DataExtractor():
-
-    def __init__(self, dataPath='', type='train'):
-
-        self.type = type
-        self.rootPath = dataPath
-
-    def GetData(self):
-
-        if self.type == "train":
-            return self.__GetTrainData()
-        if self.type == "valid":
-            return self.__GetValidData()
-        if self.type == "test":
-            return self.__GetTestData()
-
-
-    def __GetTrainData(self):
-
-        pathList = []
-
-        root = os.path.join(self.rootPath, 'train')
-        fileNames = os.listdir(root)
-        fileNames.sort()
-        for name in fileNames:
-            path = os.path.join(root, name)
-            pathList.append(path)
-
-        return pathList
-
-    def __GetValidData(self):
-
-        pathList = []
-
-        root = os.path.join(self.rootPath, 'valid')
-        fileNames = os.listdir(root)
-        fileNames.sort()
-        for name in fileNames:
-            path = os.path.join(root, name)
-            pathList.append(path)
-
-        return pathList
-
-    def __GetTestData(self):
-
-        pathList = []
-
-        root = os.path.join(self.rootPath, 'test')
-        fileNames = os.listdir(root)
-        fileNames.sort()
-        for name in fileNames:
-            path = os.path.join(root, name)
-            pathList.append(path)
-
-        return pathList
-

spikezoo/archs/spikeformer/DataProcess/DataLoader.py

@@ -1,115 +0,0 @@
-import os
-import torch
-# from torchvision import transforms
-from torch.utils import data
-import numpy as np
-from PIL import Image
-import cv2
-import random
-
-
-from DataProcess.DataExtactor import DataExtractor
-from DataProcess.LoadSpike import LoadSpike, load_spike_raw
-
-class Dataset(data.Dataset):
-
-    def __init__(self, pathList, dataType, spikeRadius):
-
-        self.pathList = pathList
-        self.dataType = dataType
-        self.spikeRadius = spikeRadius
-
-        #Random Rotation
-        if self.dataType == "train":
-            self.choice = [0, 1, 2, 3]
-        else:
-            self.choice = [0]
-
-    def __getitem__(self, index):
-
-        spSeq, gtFrames = self.GetItem(index)
-
-        return spSeq, gtFrames
-
-    def __len__(self):
-
-        return len(self.pathList)
-
-    def GetItem(self, index):
-
-        path = self.pathList[index]
-        spSeq, gtFrames = LoadSpike(path)
-
-        spLen, _, _ = spSeq.shape
-        gtLen, _, _ = gtFrames.shape
-        spCenter = spLen // 2
-        gtCenter = gtLen // 2
-
-        spLeft, spRight = (spCenter - self.spikeRadius,
-                           spCenter + self.spikeRadius)
-        spRight = spRight + 1
-        spSeq = spSeq[spLeft:spRight]
-
-        gtFrame = gtFrames[gtCenter]
-
-        spSeq = np.pad(spSeq, ((0, 0), (3, 3), (0, 0)), mode='constant')
-        spSeq = spSeq.astype(float) * 2 - 1
-
-        gtFrame = gtFrame.astype(float) / 255. * 2.0 - 1.
-
-
-        spSeq = torch.FloatTensor(spSeq)
-        gtFrame = torch.FloatTensor(gtFrame)
-
-        '''
-        Rotate the spike frame and Gt frame by ramdom degree, 
-        depending on the values of 'self.choice'
-        '''
-        # choice = random.choice(self.choice)
-        # spSeq = torch.rot90(spSeq, choice, dims=(1,2))
-        # gtFrame =torch.rot90(gtFrame, choice, dims=(1,2))
-        return spSeq, gtFrame
-
-
-
-
-
-class DataContainer():
-
-    def __init__(self, dataPath='', dataType='train',
-                 spikeRadius=16, batchSize=128, numWorks=0):
-
-        self.dataPath = dataPath
-        self.dataType = dataType
-        self.spikeRadius = spikeRadius
-        self.batchSize = batchSize
-        self.numWorks = numWorks
-
-        self.__GetData()
-
-    def __GetData(self):
-
-        dataset = None
-
-        dataset = DataExtractor(dataPath=self.dataPath, type=self.dataType)
-        self.pathList = dataset.GetData()
-
-    def GetLoader(self):
-
-        dataset = Dataset(self.pathList, self.dataType, self.spikeRadius)
-        dataLoader = None
-        if self.dataType == "train":
-            dataLoader = data.DataLoader(dataset, batch_size=self.batchSize, shuffle=True,
-                                         num_workers=self.numWorks, pin_memory=False)
-        else:
-            dataLoader = data.DataLoader(dataset, batch_size=self.batchSize, shuffle=False,
-                                         num_workers=self.numWorks, pin_memory=False)
-
-        return dataLoader
-
-if __name__ == "__main__":
-
-    pass
-
-
-

spikezoo/archs/spikeformer/DataProcess/LoadSpike.py

@@ -1,39 +0,0 @@
-import numpy as np
-
-def load_spike_numpy(path: str) -> (np.ndarray, np.ndarray):
-    '''
-    Load a spike sequence with it's tag from prepacked `.npz` file.\n
-    The sequence is of shape (`length`, `height`, `width`) and tag of
-        shape (`height`, `width`).
-    '''
-    data = np.load(path)
-    seq, tag, length = data['seq'], data['tag'], int(data['length'])
-    seq = np.array([(seq[i // 8] >> (i & 7)) & 1 for i in range(length)])
-    return seq, tag
-
-def LoadSpike(path: str) -> (np.ndarray, np.ndarray):
-    '''
-    Load a spike sequence,  the corresponding ground-truth frame sequence,
-    and sequence length.
-    spSeq: an ndarray of shape('sequence number', 'height', 'width')
-    gtFrames: an ndarray of shape('sequence length', 'height', 'width')
-    '''
-    data = np.load(path)
-    spSeq, gtFrames, length = data['spSeq'], data['gt'], int(data['length'])
-    spSeq = np.array([(spSeq[i // 8] >> (i & 7)) & 1 for i in range(length)])
-    return spSeq, gtFrames
-
-def load_spike_raw(path: str, width=400, height=250) -> np.ndarray:
-    '''
-    Load bit-compact raw spike data into an ndarray of shape
-        (`sequence length`, `height`, `width`).
-    '''
-    with open(path, 'rb') as f:
-        fbytes = f.read()
-    fnum = (len(fbytes) * 8) // (width * height)  # number of frames
-    frames = np.frombuffer(fbytes, dtype=np.uint8)
-    frames = np.array([frames & (1 << i) for i in range(8)])
-    frames = frames.astype(np.bool).astype(np.uint8)
-    frames = frames.transpose(1, 0).reshape(fnum, height, width)
-    frames = np.flip(frames, 1)
-    return frames

spikezoo/archs/spikeformer/EvalResults/readme

@@ -1 +0,0 @@
-This is a folder for saving the images reconstructed from validation/testing set !

spikezoo/archs/spikeformer/LICENSE

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

spikezoo/archs/spikeformer/Metrics/Metrics.py

@@ -1,50 +0,0 @@
-import numpy as np
-from skimage import metrics
-
-class Metrics():
-
-    def __init__(self):
-        self.best_psnr = 0.
-        self.best_ssim = 0.
-        self.best_niqe = 0.
-
-    def Update(self, psnr=0., ssim=0., niqe=0.):
-        self.best_psnr = psnr
-        self.best_ssim = ssim
-        self.best_niqe = niqe
-
-    def GetBestMetrics(self):
-
-        return self.best_psnr, self.best_ssim, self.best_niqe
-
-    def Cal_PSNR(self, preImgs, gtImgs): #shape:[B, H, W]
-
-        B, _, _ = preImgs.shape
-        total_psnr = 0.
-        for i, (pre, gt) in enumerate(zip(preImgs, gtImgs)):
-            print(i+1, metrics.peak_signal_noise_ratio(gt, pre))
-            total_psnr += metrics.peak_signal_noise_ratio(gt, pre)
-
-        avg_psnr = total_psnr / B
-
-        return avg_psnr
-
-    def Cal_SSIM(self, preImgs, gtImgs): #shape:[B, H, W]
-
-        B, _, _ = preImgs.shape
-        total_ssim = 0.
-        for i, (pre, gt) in enumerate(zip(preImgs, gtImgs)):
-            total_ssim += metrics.structural_similarity(pre, gt)
-
-        avg_ssim = total_ssim / B
-
-        return avg_ssim
-
-
-if __name__ == "__main__":
-
-    a = np.random.random((2,256,256))
-    b = np.random.random((2,256,256))
-    metrics = Metrics()
-
-    print(metrics.Cal_NIQE(a))

spikezoo/archs/spikeformer/Model/Loss.py

@@ -1,89 +0,0 @@
-import torch
-import torch.nn as nn
-import torch.nn.functional as F
-
-class CharbonnierLoss(nn.Module):
-    """Charbonnier Loss (L1)"""
-
-    def __init__(self, eps=1e-3):
-        super(CharbonnierLoss, self).__init__()
-        self.eps = eps
-
-    def forward(self, x, y):
-        diff = x - y
-        # loss = torch.sum(torch.sqrt(diff * diff + self.eps))
-        loss = torch.mean(torch.sqrt((diff * diff) + (self.eps*self.eps)))
-        return loss
-
-class EdgeLoss(nn.Module):
-    def __init__(self):
-        super(EdgeLoss, self).__init__()
-        k = torch.Tensor([[.05, .25, .4, .25, .05]])
-        # self.kernel = torch.matmul(k.t(),k).unsqueeze(0).repeat(3,1,1,1)
-        self.kernel = torch.matmul(k.t(),k).unsqueeze(0).unsqueeze(0).repeat(1,3,1,1) #这个的repeat也是后加的
-        # print(self.kernel.shape)
-        if torch.cuda.is_available():
-            self.kernel = self.kernel.cuda()
-        self.loss = CharbonnierLoss()
-
-    def conv_gauss(self, img):
-        # print('aaaa')
-        # print(img.shape)
-        n_channels, _, kw, kh = self.kernel.shape
-        img = F.pad(img, (kw//2, kh//2, kw//2, kh//2), mode='replicate')
-        # return F.conv2d(img, self.kernel, groups=n_channels)
-        return F.conv2d(img, self.kernel)
-
-    def laplacian_kernel(self, current):
-        filtered    = self.conv_gauss(current)    # filter
-        down        = filtered[:,:,::2,::2]               # downsample
-        new_filter  = torch.zeros_like(filtered)
-        new_filter[:,:,::2,::2] = down*4                  # upsample
-        filtered    = self.conv_gauss(new_filter.repeat(1,3,1,1)) # filter  #这里为什么需要repeat一下？原文的目的是什么？否则不能正常运行
-        diff = current - filtered
-        return diff
-
-    def forward(self, x, y):
-        y = y.repeat(1,3,1,1)
-        x = x.repeat(1,3,1,1)
-        # print('bbbbbb')
-        # print(x.shape)
-        # print(y.shape)
-        loss = self.loss(self.laplacian_kernel(x), self.laplacian_kernel(y))
-        return loss
-
-
-class VGGLoss4(nn.Module):
-    def __init__(self, path: str):
-        super().__init__()
-        self.features = nn.Sequential(
-            nn.Conv2d(3, 64, 3, 1, 1),
-            nn.ReLU(inplace=True),
-            nn.Conv2d(64, 64, 3, 1, 1),
-            nn.ReLU(inplace=True),
-            nn.MaxPool2d(2),
-            nn.Conv2d(64, 128, 3, 1, 1),
-            nn.ReLU(inplace=True),
-            nn.Conv2d(128, 128, 3, 1, 1),
-            nn.ReLU(inplace=True),
-            nn.MaxPool2d(2),
-            nn.Conv2d(128, 256, 3, 1, 1),
-            nn.ReLU(inplace=True),
-            # nn.Conv2d(256, 256, 3, 1, 1),
-            # nn.ReLU(inplace=True),
-            # nn.Conv2d(256, 256, 3, 1, 1),
-            # nn.ReLU(inplace=True),
-        )
-        self.load_state_dict(torch.load(path))
-        for param in self.parameters():
-            param.requires_grad = False
-
-    def forward(self, real_y, fake_y):
-        real_y = real_y.repeat((1, 3, 1, 1))
-        fake_y = fake_y.repeat((1, 3, 1, 1))
-        with torch.no_grad():
-            real_f = self.features(real_y)
-        fake_f = self.features(fake_y)
-        return F.mse_loss(real_f, fake_f)
-
-