diff --git a/__pycache__/config.cpython-311.pyc b/__pycache__/config.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..8cca86d7e2eb9fe1da7b171985ca2d4c7593f438 Binary files /dev/null and b/__pycache__/config.cpython-311.pyc differ diff --git a/__pycache__/dataset.cpython-311.pyc b/__pycache__/dataset.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..7c71a7d93bbbf580a3d0798fc235f20021395f26 Binary files /dev/null and b/__pycache__/dataset.cpython-311.pyc differ diff --git a/__pycache__/discriminator_model.cpython-311.pyc b/__pycache__/discriminator_model.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..7ed649cff9ff4ae61eddbf4bc3d7525a21f0c35a Binary files /dev/null and b/__pycache__/discriminator_model.cpython-311.pyc differ diff --git a/__pycache__/generator_model.cpython-311.pyc b/__pycache__/generator_model.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..e09e73746d7fde7933d628f7bf999287e05182b8 Binary files /dev/null and b/__pycache__/generator_model.cpython-311.pyc differ diff --git a/__pycache__/train.cpython-311.pyc b/__pycache__/train.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..974c7da56560f665b0cf0edab01363eeb257e9ab Binary files /dev/null and b/__pycache__/train.cpython-311.pyc differ diff --git a/__pycache__/utils.cpython-311.pyc b/__pycache__/utils.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..539deda84c6fa4e7dae88e701159ace744de5922 Binary files /dev/null and b/__pycache__/utils.cpython-311.pyc differ diff --git a/config.py b/config.py new file mode 100644 index 0000000000000000000000000000000000000000..24ef9985271ed5b920e0f982a014c9dbefdbcb94 --- /dev/null +++ b/config.py @@ -0,0 +1,38 @@ +import albumentations as A +from albumentations.pytorch import ToTensorV2 + +DEVICE = "cuda" +TRAIN_DIR = "data/train" +VAL_DIR = "data/val" +LEARNING_RATE = 2e-4 +BATCH_SIZE = 16 +NUM_WORKERS = 2 +IMAGE_SIZE = 256 +CHANNELS_IMG = 3 +L1_LAMBDA = 100 +LAMBDA_GP = 10 +NUM_EPOCHS = 1 +LOAD_MODEL = True +SAVE_MODEL = False +CHECKPOINT_DISC = "disc.pth.tar" +CHECKPOINT_GEN = "gen.pth.tar" + +both_transform = A.Compose( + [A.Resize(width=256, height=256), ], additional_targets={"image0": "image"}, is_check_shapes=False +) + +transform_only_input = A.Compose( + [ + A.HorizontalFlip(p=0.5), + A.ColorJitter(p=0.2), + A.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5], max_pixel_value=255.0, ), + ToTensorV2(), + ] +) + +transform_only_mask = A.Compose( + [ + A.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5], max_pixel_value=255.0, ), + ToTensorV2(), + ] +) \ No newline at end of file diff --git a/data/train/comics/1002.jpg b/data/train/comics/1002.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5af06b5fc01b8a3ecc0c888b60367bb21ea5d347 Binary files /dev/null and b/data/train/comics/1002.jpg differ diff --git a/data/train/comics/1130.jpg b/data/train/comics/1130.jpg new file mode 100644 index 0000000000000000000000000000000000000000..0a140a2a5354d02ce7a16795554e64ccb769ae71 Binary files /dev/null and b/data/train/comics/1130.jpg differ diff --git a/data/train/comics/1189.jpg b/data/train/comics/1189.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ad2ebf1ccf22957f4355ce396e49101672c8b27b Binary files /dev/null and b/data/train/comics/1189.jpg differ diff --git a/data/train/comics/1218.jpg b/data/train/comics/1218.jpg new file mode 100644 index 0000000000000000000000000000000000000000..25ccb6825f0e06a62b61c7753c5e8f2fbb00ad6e Binary files /dev/null and b/data/train/comics/1218.jpg differ diff --git a/data/train/comics/1273.jpg b/data/train/comics/1273.jpg new file mode 100644 index 0000000000000000000000000000000000000000..36ea69a6ded180bc3663ce6eaa2773390e1c9fa3 Binary files /dev/null and b/data/train/comics/1273.jpg differ diff --git a/data/train/comics/1300.jpg b/data/train/comics/1300.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d529410d755f4a2da5eeeb0a8533126167b15d95 Binary files /dev/null and b/data/train/comics/1300.jpg differ diff --git a/data/train/comics/1339.jpg b/data/train/comics/1339.jpg new file mode 100644 index 0000000000000000000000000000000000000000..73a64c4b035972a50f3ff534230aae18ad58ca79 Binary files /dev/null and b/data/train/comics/1339.jpg differ diff --git a/data/train/comics/1370.jpg b/data/train/comics/1370.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8f7b3e8354e52145a6c3dd790fbf650839778600 Binary files /dev/null and b/data/train/comics/1370.jpg differ diff --git a/data/train/comics/1460.jpg b/data/train/comics/1460.jpg new file mode 100644 index 0000000000000000000000000000000000000000..351f8ba15086ba363dae995c2de84e9df6dc078f Binary files /dev/null and b/data/train/comics/1460.jpg differ diff --git a/data/train/comics/1471.jpg b/data/train/comics/1471.jpg new file mode 100644 index 0000000000000000000000000000000000000000..44e72114c95b663d45fe07ae77d0a67c4e1cd909 Binary files /dev/null and b/data/train/comics/1471.jpg differ diff --git a/data/train/comics/1479.jpg b/data/train/comics/1479.jpg new file mode 100644 index 0000000000000000000000000000000000000000..033b735418b646f06cbb1761e1d4eef2f9ce6a2e Binary files /dev/null and b/data/train/comics/1479.jpg differ diff --git a/data/train/comics/1804.jpg b/data/train/comics/1804.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8faf1c1af22fa4cdffe9fd2b983eee40d8a14417 Binary files /dev/null and b/data/train/comics/1804.jpg differ diff --git a/data/train/comics/1862.jpg b/data/train/comics/1862.jpg new file mode 100644 index 0000000000000000000000000000000000000000..760ab9a5da0f5bbfccae45339e0b7e5395fcaefe Binary files /dev/null and b/data/train/comics/1862.jpg differ diff --git a/data/train/comics/1927.jpg b/data/train/comics/1927.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6c81ae8a63fcea0c23c0af6e5850b05c179e4148 Binary files /dev/null and b/data/train/comics/1927.jpg differ diff --git a/data/train/comics/2078.jpg b/data/train/comics/2078.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d245fab0e5c3dcd899bb3b9904fec3885cb25af7 Binary files /dev/null and b/data/train/comics/2078.jpg differ diff --git a/data/train/comics/2101.jpg b/data/train/comics/2101.jpg new file mode 100644 index 0000000000000000000000000000000000000000..a9fe5232ae0ca4ef723d5be686b2261cd05f5ddc Binary files /dev/null and b/data/train/comics/2101.jpg differ diff --git a/data/train/comics/2131.jpg b/data/train/comics/2131.jpg new file mode 100644 index 0000000000000000000000000000000000000000..93db3e858ebcdf864e0269747f8a571b603860ee Binary files /dev/null and b/data/train/comics/2131.jpg differ diff --git a/data/train/comics/2231.jpg b/data/train/comics/2231.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5db9185a091cf01d2730034c8300596d8de3a358 Binary files /dev/null and b/data/train/comics/2231.jpg differ diff --git a/data/train/comics/2355.jpg b/data/train/comics/2355.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1f254fa1b1a8ed70c9584f1badf998af9da700df Binary files /dev/null and b/data/train/comics/2355.jpg differ diff --git a/data/train/comics/248.jpg b/data/train/comics/248.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7c0833de2ccd25233b2b261060c5156d5f9bc1dd Binary files /dev/null and b/data/train/comics/248.jpg differ diff --git a/data/train/comics/2508.jpg b/data/train/comics/2508.jpg new file mode 100644 index 0000000000000000000000000000000000000000..13b86497fd92191da4275dfa24578a805702b283 Binary files /dev/null and b/data/train/comics/2508.jpg differ diff --git a/data/train/comics/2514.jpg b/data/train/comics/2514.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2811521641222e270365f85bb8dffcf35af7e881 Binary files /dev/null and b/data/train/comics/2514.jpg differ diff --git a/data/train/comics/2772.jpg b/data/train/comics/2772.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3fad24ab3a7a09492a8587e4c12be8241d974c81 Binary files /dev/null and b/data/train/comics/2772.jpg differ diff --git a/data/train/comics/2857.jpg b/data/train/comics/2857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9649a69e8f8c9cfa887615f80f088383d0e379b8 Binary files /dev/null and b/data/train/comics/2857.jpg differ diff --git a/data/train/comics/312.jpg b/data/train/comics/312.jpg new file mode 100644 index 0000000000000000000000000000000000000000..52d8fb867eb9e7b4f73e1226ddcef00a5a9a7fab Binary files /dev/null and b/data/train/comics/312.jpg differ diff --git a/data/train/comics/3223.jpg b/data/train/comics/3223.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6bb1aebd5035d280ed6a753626e08597edf21b29 Binary files /dev/null and b/data/train/comics/3223.jpg differ diff --git a/data/train/comics/3322.jpg b/data/train/comics/3322.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1e72630d1f60de4c679d0317054ed4a6321ce182 Binary files /dev/null and b/data/train/comics/3322.jpg differ diff --git a/data/train/comics/3359.jpg b/data/train/comics/3359.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2fa644e58d070b6e19f12aa3fe6f2d56889b2dbd Binary files /dev/null and b/data/train/comics/3359.jpg differ diff --git a/data/train/comics/3428.jpg b/data/train/comics/3428.jpg new file mode 100644 index 0000000000000000000000000000000000000000..26a5beb6d41360b16ea3834c61fc1f4a71debf4a Binary files /dev/null and b/data/train/comics/3428.jpg differ diff --git a/data/train/comics/3520.jpg b/data/train/comics/3520.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d078a249d2dca1e285b5acf246c08c9821ac9655 Binary files /dev/null and b/data/train/comics/3520.jpg differ diff --git a/data/train/comics/3592.jpg b/data/train/comics/3592.jpg new file mode 100644 index 0000000000000000000000000000000000000000..522196b28d05107ab768c7d690f48b2408e829dd Binary files /dev/null and b/data/train/comics/3592.jpg differ diff --git a/data/train/comics/3733.jpg b/data/train/comics/3733.jpg new file mode 100644 index 0000000000000000000000000000000000000000..db991958f8d5549240799e584de47e6b3057acbd Binary files /dev/null and b/data/train/comics/3733.jpg differ diff --git a/data/train/comics/3876.jpg b/data/train/comics/3876.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3d7217d92c2ecc186214811e94230d97069a2f3c Binary files /dev/null and b/data/train/comics/3876.jpg differ diff --git a/data/train/comics/3970.jpg b/data/train/comics/3970.jpg new file mode 100644 index 0000000000000000000000000000000000000000..a816f4ddf35a28c5671627471c4f59fe3155a6a7 Binary files /dev/null and b/data/train/comics/3970.jpg differ diff --git a/data/train/comics/3978.jpg b/data/train/comics/3978.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2e91b0c7d4e4300289f73415aed4e09d1b08a42d Binary files /dev/null and b/data/train/comics/3978.jpg differ diff --git a/data/train/comics/40.jpg b/data/train/comics/40.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e2edd6de1def1c630dfb98204b7b9598bd8ed5ac Binary files /dev/null and b/data/train/comics/40.jpg differ diff --git a/data/train/comics/4075.jpg b/data/train/comics/4075.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7c0f6407f61cd6c1d514185917f487aa5b15ecc9 Binary files /dev/null and b/data/train/comics/4075.jpg differ diff --git a/data/train/comics/4081.jpg b/data/train/comics/4081.jpg new file mode 100644 index 0000000000000000000000000000000000000000..18121bab9426c48d8ca01bb791e7a672039eb44a Binary files /dev/null and b/data/train/comics/4081.jpg differ diff --git a/data/train/comics/416.jpg b/data/train/comics/416.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d10eafbf094c7e42a4d3d7aaafc2ee7cb3c53657 Binary files /dev/null and b/data/train/comics/416.jpg differ diff --git a/data/train/comics/4183.jpg b/data/train/comics/4183.jpg new file mode 100644 index 0000000000000000000000000000000000000000..eb52f0b93e40a5d22d98be81427c9b6051067fdd Binary files /dev/null and b/data/train/comics/4183.jpg differ diff --git a/data/train/comics/4368.jpg b/data/train/comics/4368.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f7aa7ea5d9b70185b7bc4b202d913f5439621f94 Binary files /dev/null and b/data/train/comics/4368.jpg differ diff --git a/data/train/comics/4613.jpg b/data/train/comics/4613.jpg new file mode 100644 index 0000000000000000000000000000000000000000..66f2a805b72abb7e0fd21e0afb9bf80bea804760 Binary files /dev/null and b/data/train/comics/4613.jpg differ diff --git a/data/train/comics/4660.jpg b/data/train/comics/4660.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ed4c6e913eecb8b862d81e035b8fe500d51e5910 Binary files /dev/null and b/data/train/comics/4660.jpg differ diff --git a/data/train/comics/4682.jpg b/data/train/comics/4682.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5a03eea276608d4d8b784e17ea17f1b9ea41321f Binary files /dev/null and b/data/train/comics/4682.jpg differ diff --git a/data/train/comics/4765.jpg b/data/train/comics/4765.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c3ffdd56a62f6dee785b92c96fa0fb1723a2e319 Binary files /dev/null and b/data/train/comics/4765.jpg differ diff --git a/data/train/comics/48.jpg b/data/train/comics/48.jpg new file mode 100644 index 0000000000000000000000000000000000000000..943da05be258d5444628834b96ec8352e457041e Binary files /dev/null and b/data/train/comics/48.jpg differ diff --git a/data/train/comics/5093.jpg b/data/train/comics/5093.jpg new file mode 100644 index 0000000000000000000000000000000000000000..14e83f51e440d74e231fd387b93233e09c46e57a Binary files /dev/null and b/data/train/comics/5093.jpg differ diff --git a/data/train/comics/5368.jpg b/data/train/comics/5368.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1b4634f2f9ab7012b8772361081765e1154b4dba Binary files /dev/null and b/data/train/comics/5368.jpg differ diff --git a/data/train/comics/5456.jpg b/data/train/comics/5456.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3e88276e647f3e6b3d263a9e08601bfbf994b074 Binary files /dev/null and b/data/train/comics/5456.jpg differ diff --git a/data/train/comics/5528.jpg b/data/train/comics/5528.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ffaa0f5cf2b5676813fde31414ff2c17b5495afd Binary files /dev/null and b/data/train/comics/5528.jpg differ diff --git a/data/train/comics/5831.jpg b/data/train/comics/5831.jpg new file mode 100644 index 0000000000000000000000000000000000000000..dfeb5cc7554e591efc523631a69d4ec474f78bee Binary files /dev/null and b/data/train/comics/5831.jpg differ diff --git a/data/train/comics/6165.jpg b/data/train/comics/6165.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d6900eab53c14a79c1554b7b8c14e5695ccfa078 Binary files /dev/null and b/data/train/comics/6165.jpg differ diff --git a/data/train/comics/6380.jpg b/data/train/comics/6380.jpg new file mode 100644 index 0000000000000000000000000000000000000000..681e8944c90ef84cfe3317a3279f6274e9216f54 Binary files /dev/null and b/data/train/comics/6380.jpg differ diff --git a/data/train/comics/6431.jpg b/data/train/comics/6431.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c97c7ce3bfcd2dc433eb5700f270eace0876bc2f Binary files /dev/null and b/data/train/comics/6431.jpg differ diff --git a/data/train/comics/6440.jpg b/data/train/comics/6440.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7fe37bb6d7bfb0237ae1a6686f157e17f48e577c Binary files /dev/null and b/data/train/comics/6440.jpg differ diff --git a/data/train/comics/6493.jpg b/data/train/comics/6493.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e3b85f4e7001df95e495c28446fb78a7d96ae5f9 Binary files /dev/null and b/data/train/comics/6493.jpg differ diff --git a/data/train/comics/6495.jpg b/data/train/comics/6495.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6fe581025aaa62079ef7bfcc2474123a38f04634 Binary files /dev/null and b/data/train/comics/6495.jpg differ diff --git a/data/train/comics/6542.jpg b/data/train/comics/6542.jpg new file mode 100644 index 0000000000000000000000000000000000000000..fdf8f9e8a4321a6d72b1f9759aa190c9b31248d7 Binary files /dev/null and b/data/train/comics/6542.jpg differ diff --git a/data/train/comics/6573.jpg b/data/train/comics/6573.jpg new file mode 100644 index 0000000000000000000000000000000000000000..74764d8f62a968ec4b1005c1983cca77669f5406 Binary files /dev/null and b/data/train/comics/6573.jpg differ diff --git a/data/train/comics/6676.jpg b/data/train/comics/6676.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e154fc985277a4713c241841510293cb118c390e Binary files /dev/null and b/data/train/comics/6676.jpg differ diff --git a/data/train/comics/6781.jpg b/data/train/comics/6781.jpg new file mode 100644 index 0000000000000000000000000000000000000000..74d13a38f830f21ea210a9bc21d0599911038923 Binary files /dev/null and b/data/train/comics/6781.jpg differ diff --git a/data/train/comics/6806.jpg b/data/train/comics/6806.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9e119ba5d365a7dd43789b28df73bdb9964c4d14 Binary files /dev/null and b/data/train/comics/6806.jpg differ diff --git a/data/train/comics/6857.jpg b/data/train/comics/6857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e2212d546fdaea4799d1064435535e42019d464c Binary files /dev/null and b/data/train/comics/6857.jpg differ diff --git a/data/train/comics/6920.jpg b/data/train/comics/6920.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7bea8b09420ccfa67ad70e5d25c6aecc85da7cf8 Binary files /dev/null and b/data/train/comics/6920.jpg differ diff --git a/data/train/comics/70.jpg b/data/train/comics/70.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5bd324d8a30b7503cfd6bc7da2421ca523945e86 Binary files /dev/null and b/data/train/comics/70.jpg differ diff --git a/data/train/comics/7128.jpg b/data/train/comics/7128.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ec78d95f4040fea214e4a3d6861beec70e26c3c0 Binary files /dev/null and b/data/train/comics/7128.jpg differ diff --git a/data/train/comics/7209.jpg b/data/train/comics/7209.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6301bbb8eba4d9e4742490c47aa6bc4a1cd45b38 Binary files /dev/null and b/data/train/comics/7209.jpg differ diff --git a/data/train/comics/7299.jpg b/data/train/comics/7299.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ff279f015b17d3c729fe7061f37f0393e62af1f5 Binary files /dev/null and b/data/train/comics/7299.jpg differ diff --git a/data/train/comics/7321.jpg b/data/train/comics/7321.jpg new file mode 100644 index 0000000000000000000000000000000000000000..87a634ce55090d697989f8c6ef6e85acf59a8f7d Binary files /dev/null and b/data/train/comics/7321.jpg differ diff --git a/data/train/comics/7721.jpg b/data/train/comics/7721.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c8df28c62a6c38cccffdb2347dec84ba360561ee Binary files /dev/null and b/data/train/comics/7721.jpg differ diff --git a/data/train/comics/7823.jpg b/data/train/comics/7823.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8302ecfb9a812bd683b40a181fc9266bfdf99464 Binary files /dev/null and b/data/train/comics/7823.jpg differ diff --git a/data/train/comics/7853.jpg b/data/train/comics/7853.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b67755b67d551d99028a5d5d7f1dd417f388ea68 Binary files /dev/null and b/data/train/comics/7853.jpg differ diff --git a/data/train/comics/7930.jpg b/data/train/comics/7930.jpg new file mode 100644 index 0000000000000000000000000000000000000000..870c5bb35e2c9b03eb4b1271d7e1aa74731bc966 Binary files /dev/null and b/data/train/comics/7930.jpg differ diff --git a/data/train/comics/7955.jpg b/data/train/comics/7955.jpg new file mode 100644 index 0000000000000000000000000000000000000000..610f889d59dca9df648cc3074a0b4b210b30a18d Binary files /dev/null and b/data/train/comics/7955.jpg differ diff --git a/data/train/comics/8127.jpg b/data/train/comics/8127.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cbc767ffec72d4aa8dd36685855f4e901ee9f73d Binary files /dev/null and b/data/train/comics/8127.jpg differ diff --git a/data/train/comics/8174.jpg b/data/train/comics/8174.jpg new file mode 100644 index 0000000000000000000000000000000000000000..19a0487ea9a6e657ae863e164cb78ea2b241dc42 Binary files /dev/null and b/data/train/comics/8174.jpg differ diff --git a/data/train/comics/8251.jpg b/data/train/comics/8251.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1c50c717654d23a55dc9bdfc7c5e0437940b9e6b Binary files /dev/null and b/data/train/comics/8251.jpg differ diff --git a/data/train/comics/8314.jpg b/data/train/comics/8314.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6e512a31513f006b5e3032ee99ebbc2c04c4fc83 Binary files /dev/null and b/data/train/comics/8314.jpg differ diff --git a/data/train/comics/833.jpg b/data/train/comics/833.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4d15b5b8d8a820fe2ebbda2def1a5a1eace54094 Binary files /dev/null and b/data/train/comics/833.jpg differ diff --git a/data/train/comics/8332.jpg b/data/train/comics/8332.jpg new file mode 100644 index 0000000000000000000000000000000000000000..765150e6404775f9cfba6ad001b26cee72927731 Binary files /dev/null and b/data/train/comics/8332.jpg differ diff --git a/data/train/comics/8390.jpg b/data/train/comics/8390.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8383cc8d91aa87731cc5318f7ea501bd71c4f84b Binary files /dev/null and b/data/train/comics/8390.jpg differ diff --git a/data/train/comics/8448.jpg b/data/train/comics/8448.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f4905e7f3afdaa9202c5efd143aeac5de483fe42 Binary files /dev/null and b/data/train/comics/8448.jpg differ diff --git a/data/train/comics/8559.jpg b/data/train/comics/8559.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ab2267f1bea0bbc4ed58716afa7c093d1cd504d9 Binary files /dev/null and b/data/train/comics/8559.jpg differ diff --git a/data/train/comics/8720.jpg b/data/train/comics/8720.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4b921e3b5091ac15b3a82b5d7d5b6606b75e0077 Binary files /dev/null and b/data/train/comics/8720.jpg differ diff --git a/data/train/comics/8775.jpg b/data/train/comics/8775.jpg new file mode 100644 index 0000000000000000000000000000000000000000..63af05582b2e75a9df05327ef86f53eb4a081ba0 Binary files /dev/null and b/data/train/comics/8775.jpg differ diff --git a/data/train/comics/8834.jpg b/data/train/comics/8834.jpg new file mode 100644 index 0000000000000000000000000000000000000000..0a54130d809de92bf29a951eadb509100cd96618 Binary files /dev/null and b/data/train/comics/8834.jpg differ diff --git a/data/train/comics/8838.jpg b/data/train/comics/8838.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ab2813c48f820068a058afd62869fdc8b152ec0c Binary files /dev/null and b/data/train/comics/8838.jpg differ diff --git a/data/train/comics/8857.jpg b/data/train/comics/8857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7b89a33f3feb4efbceaa53996bd22793f641eeb3 Binary files /dev/null and b/data/train/comics/8857.jpg differ diff --git a/data/train/comics/8884.jpg b/data/train/comics/8884.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2bd6603c71913a48f7facd998630c5bbafe46b0c Binary files /dev/null and b/data/train/comics/8884.jpg differ diff --git a/data/train/comics/8910.jpg b/data/train/comics/8910.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4bf4b2771cf0a941ea075b782f099d609d71be7d Binary files /dev/null and b/data/train/comics/8910.jpg differ diff --git a/data/train/comics/8998.jpg b/data/train/comics/8998.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bad044dfd36d146c2508371a0b2a2e35feaad64d Binary files /dev/null and b/data/train/comics/8998.jpg differ diff --git a/data/train/comics/9040.jpg b/data/train/comics/9040.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1c94a4e54f8834cad0099301e01c85e7e9f0a402 Binary files /dev/null and b/data/train/comics/9040.jpg differ diff --git a/data/train/comics/9125.jpg b/data/train/comics/9125.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ab66a89442330c3ef4fc7de66bb0d76bc7ea06cc Binary files /dev/null and b/data/train/comics/9125.jpg differ diff --git a/data/train/comics/9160.jpg b/data/train/comics/9160.jpg new file mode 100644 index 0000000000000000000000000000000000000000..04d77c79bc1c04dbb8d69aa3137abd37a180ccc4 Binary files /dev/null and b/data/train/comics/9160.jpg differ diff --git a/data/train/comics/9173.jpg b/data/train/comics/9173.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ddb0224e2f9e9e54abf5a0da31f3a57f9795b32e Binary files /dev/null and b/data/train/comics/9173.jpg differ diff --git a/data/train/comics/9608.jpg b/data/train/comics/9608.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ce2d0d77f4ad068901dcdc6e89288de26e560523 Binary files /dev/null and b/data/train/comics/9608.jpg differ diff --git a/data/train/comics/9706.jpg b/data/train/comics/9706.jpg new file mode 100644 index 0000000000000000000000000000000000000000..01e524794c7b60bb08960d1f3d82f7d00b6aa5fb Binary files /dev/null and b/data/train/comics/9706.jpg differ diff --git a/data/train/comics/9723.jpg b/data/train/comics/9723.jpg new file mode 100644 index 0000000000000000000000000000000000000000..24f563e02edec5aeba702da4482b60c19ffa62e7 Binary files /dev/null and b/data/train/comics/9723.jpg differ diff --git a/data/train/comics/9982.jpg b/data/train/comics/9982.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bfde43c299b88e210b6ea13ce6ed3f798ad89b39 Binary files /dev/null and b/data/train/comics/9982.jpg differ diff --git a/data/train/comics/9985.jpg b/data/train/comics/9985.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6788bc9ea81de75238b0917ef5a3dd5d1c7155d2 Binary files /dev/null and b/data/train/comics/9985.jpg differ diff --git a/data/train/face/1002.jpg b/data/train/face/1002.jpg new file mode 100644 index 0000000000000000000000000000000000000000..291f0be95d6276304c38369549ca4baae1436b5e Binary files /dev/null and b/data/train/face/1002.jpg differ diff --git a/data/train/face/1130.jpg b/data/train/face/1130.jpg new file mode 100644 index 0000000000000000000000000000000000000000..18b5a44e20167924653660f7092cb33e2178275e Binary files /dev/null and b/data/train/face/1130.jpg differ diff --git a/data/train/face/1189.jpg b/data/train/face/1189.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7ee438a634e86b2a5824bbdc1bc5a28044d706c9 Binary files /dev/null and b/data/train/face/1189.jpg differ diff --git a/data/train/face/1218.jpg b/data/train/face/1218.jpg new file mode 100644 index 0000000000000000000000000000000000000000..efda784ae31d92ee31fbc70489cc5b1c7890ef3f Binary files /dev/null and b/data/train/face/1218.jpg differ diff --git a/data/train/face/1273.jpg b/data/train/face/1273.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b50c2f5caaf04689d0f013556344edc1e0c43f2b Binary files /dev/null and b/data/train/face/1273.jpg differ diff --git a/data/train/face/1300.jpg b/data/train/face/1300.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7e78b364f76e8d90560de6ffacec4c309008be32 Binary files /dev/null and b/data/train/face/1300.jpg differ diff --git a/data/train/face/1339.jpg b/data/train/face/1339.jpg new file mode 100644 index 0000000000000000000000000000000000000000..629ec646c9e126f3fab4288e55a6ebc0e061cfdd Binary files /dev/null and b/data/train/face/1339.jpg differ diff --git a/data/train/face/1370.jpg b/data/train/face/1370.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ec4ba1d4320a153138e8bc59b08b3341f0fef5bf Binary files /dev/null and b/data/train/face/1370.jpg differ diff --git a/data/train/face/1460.jpg b/data/train/face/1460.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b2c9527925919e78cd5ca35dc12b291453cb0f4a Binary files /dev/null and b/data/train/face/1460.jpg differ diff --git a/data/train/face/1471.jpg b/data/train/face/1471.jpg new file mode 100644 index 0000000000000000000000000000000000000000..34fda2617a8eef410fdc3ea268d2f2f019a19919 Binary files /dev/null and b/data/train/face/1471.jpg differ diff --git a/data/train/face/1479.jpg b/data/train/face/1479.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d2a811bb8ace310e3d93da47a707b5489f38b1d6 Binary files /dev/null and b/data/train/face/1479.jpg differ diff --git a/data/train/face/1804.jpg b/data/train/face/1804.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8bdf311fbadb123c24990cd6f826c25f8ad260d5 Binary files /dev/null and b/data/train/face/1804.jpg differ diff --git a/data/train/face/1862.jpg b/data/train/face/1862.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bbe2920a23bcd2cc7d09bf0c7062883266921d16 Binary files /dev/null and b/data/train/face/1862.jpg differ diff --git a/data/train/face/1927.jpg b/data/train/face/1927.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2befded1d3aaa02e379d544fa9800460afd26420 Binary files /dev/null and b/data/train/face/1927.jpg differ diff --git a/data/train/face/2078.jpg b/data/train/face/2078.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c0f458b4715b8b0479d5773205a599fde82942ad Binary files /dev/null and b/data/train/face/2078.jpg differ diff --git a/data/train/face/2101.jpg b/data/train/face/2101.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7c586701cd1f4f6c8249f6b3b8fcb82904907e72 Binary files /dev/null and b/data/train/face/2101.jpg differ diff --git a/data/train/face/2131.jpg b/data/train/face/2131.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f68707bfb314f58ae0fdaffd260e68028651fc7c Binary files /dev/null and b/data/train/face/2131.jpg differ diff --git a/data/train/face/2231.jpg b/data/train/face/2231.jpg new file mode 100644 index 0000000000000000000000000000000000000000..79dbb92b89dab7448f8df454dd84b06c59967896 Binary files /dev/null and b/data/train/face/2231.jpg differ diff --git a/data/train/face/2355.jpg b/data/train/face/2355.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c69be7e90923ff6ad54f71253efcb00c5929e727 Binary files /dev/null and b/data/train/face/2355.jpg differ diff --git a/data/train/face/248.jpg b/data/train/face/248.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3393eb6f0b61017199238c9958067a45e4019505 Binary files /dev/null and b/data/train/face/248.jpg differ diff --git a/data/train/face/2508.jpg b/data/train/face/2508.jpg new file mode 100644 index 0000000000000000000000000000000000000000..dd42201550bbf9dff699064d16d3c7883c510216 Binary files /dev/null and b/data/train/face/2508.jpg differ diff --git a/data/train/face/2514.jpg b/data/train/face/2514.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4e94334cdffd46edc6ca87498c932a0d67f7e87f Binary files /dev/null and b/data/train/face/2514.jpg differ diff --git a/data/train/face/2772.jpg b/data/train/face/2772.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8078ef7ef26a18130b217a31dbdf51ab3378a766 Binary files /dev/null and b/data/train/face/2772.jpg differ diff --git a/data/train/face/2857.jpg b/data/train/face/2857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1d761d86a01f57c348536961ebf4cf099181e2f0 Binary files /dev/null and b/data/train/face/2857.jpg differ diff --git a/data/train/face/312.jpg b/data/train/face/312.jpg new file mode 100644 index 0000000000000000000000000000000000000000..395a7aaf6d7441935244e16dfc39c73393a07369 Binary files /dev/null and b/data/train/face/312.jpg differ diff --git a/data/train/face/3223.jpg b/data/train/face/3223.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1c0028c2cbd9496afc5e65daa08108db4376fa2a Binary files /dev/null and b/data/train/face/3223.jpg differ diff --git a/data/train/face/3322.jpg b/data/train/face/3322.jpg new file mode 100644 index 0000000000000000000000000000000000000000..24904ad6cea2d461f42218a065796540faf4db88 Binary files /dev/null and b/data/train/face/3322.jpg differ diff --git a/data/train/face/3359.jpg b/data/train/face/3359.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cae681ba3a5905f64f5f519a8585705920c35134 Binary files /dev/null and b/data/train/face/3359.jpg differ diff --git a/data/train/face/3428.jpg b/data/train/face/3428.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e345cf918b0ccd15ccc1d54fa207ef32f4e96880 Binary files /dev/null and b/data/train/face/3428.jpg differ diff --git a/data/train/face/3520.jpg b/data/train/face/3520.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bcfe5da0db98a30142c1cea5e42d04b9589058d2 Binary files /dev/null and b/data/train/face/3520.jpg differ diff --git a/data/train/face/3592.jpg b/data/train/face/3592.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f8be893cd9f45016738e863c26752c09c57b541e Binary files /dev/null and b/data/train/face/3592.jpg differ diff --git a/data/train/face/3733.jpg b/data/train/face/3733.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5db3fbfc9ec26d91a120c44d8b225ba2f7a9cba8 Binary files /dev/null and b/data/train/face/3733.jpg differ diff --git a/data/train/face/3876.jpg b/data/train/face/3876.jpg new file mode 100644 index 0000000000000000000000000000000000000000..99b9cf1ec6c0dc68f5bc433753838b0ca4465d95 Binary files /dev/null and b/data/train/face/3876.jpg differ diff --git a/data/train/face/3970.jpg b/data/train/face/3970.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6e10853032610193f9b7fe168e515069f434937d Binary files /dev/null and b/data/train/face/3970.jpg differ diff --git a/data/train/face/3978.jpg b/data/train/face/3978.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d9024676929e37d07fe2b0b0e1fef5f1eaf96a7c Binary files /dev/null and b/data/train/face/3978.jpg differ diff --git a/data/train/face/40.jpg b/data/train/face/40.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9156f0293d58a1ffe8f2cb1772698e324be3f269 Binary files /dev/null and b/data/train/face/40.jpg differ diff --git a/data/train/face/4075.jpg b/data/train/face/4075.jpg new file mode 100644 index 0000000000000000000000000000000000000000..988ff6923ae8ef94e681e6daab1f0493f0fbb2c5 Binary files /dev/null and b/data/train/face/4075.jpg differ diff --git a/data/train/face/4081.jpg b/data/train/face/4081.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7e6afa9a94a906616da6c34382f42376adde9c51 Binary files /dev/null and b/data/train/face/4081.jpg differ diff --git a/data/train/face/416.jpg b/data/train/face/416.jpg new file mode 100644 index 0000000000000000000000000000000000000000..92193c41e32776690dcc6b4056b216e2a70f67a8 Binary files /dev/null and b/data/train/face/416.jpg differ diff --git a/data/train/face/4183.jpg b/data/train/face/4183.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c0046c18640da00bdfbced94846fc48c7ff79090 Binary files /dev/null and b/data/train/face/4183.jpg differ diff --git a/data/train/face/4368.jpg b/data/train/face/4368.jpg new file mode 100644 index 0000000000000000000000000000000000000000..150a3e6c72ae83d9b9c54ccae2ba76a265238165 Binary files /dev/null and b/data/train/face/4368.jpg differ diff --git a/data/train/face/4613.jpg b/data/train/face/4613.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3891f226fbe506c3c08bef119f1a7ccd9d5f25c7 Binary files /dev/null and b/data/train/face/4613.jpg differ diff --git a/data/train/face/4660.jpg b/data/train/face/4660.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b72a1eddf3e6459c4a10939e05057e4677203989 Binary files /dev/null and b/data/train/face/4660.jpg differ diff --git a/data/train/face/4682.jpg b/data/train/face/4682.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e92d6d384b601e5747fa05b5294fb2e8fce247d9 Binary files /dev/null and b/data/train/face/4682.jpg differ diff --git a/data/train/face/4765.jpg b/data/train/face/4765.jpg new file mode 100644 index 0000000000000000000000000000000000000000..829cce02bee92a7b62446c32ae05930c9e9cb47c Binary files /dev/null and b/data/train/face/4765.jpg differ diff --git a/data/train/face/48.jpg b/data/train/face/48.jpg new file mode 100644 index 0000000000000000000000000000000000000000..595a7e42eaef212f8cfebf19417051fbddee30b2 Binary files /dev/null and b/data/train/face/48.jpg differ diff --git a/data/train/face/5093.jpg b/data/train/face/5093.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5dd387ce943ff232557d645cd8bf8002cd4fe243 Binary files /dev/null and b/data/train/face/5093.jpg differ diff --git a/data/train/face/5368.jpg b/data/train/face/5368.jpg new file mode 100644 index 0000000000000000000000000000000000000000..da8d5f88eed1a2166e650b0c9c21cc02c0b89c13 Binary files /dev/null and b/data/train/face/5368.jpg differ diff --git a/data/train/face/5456.jpg b/data/train/face/5456.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b1dc5061a98c0cd6bb705137f3a69427233f9850 Binary files /dev/null and b/data/train/face/5456.jpg differ diff --git a/data/train/face/5528.jpg b/data/train/face/5528.jpg new file mode 100644 index 0000000000000000000000000000000000000000..55456ec22a7686992014af348e4e70c1c7222765 Binary files /dev/null and b/data/train/face/5528.jpg differ diff --git a/data/train/face/5831.jpg b/data/train/face/5831.jpg new file mode 100644 index 0000000000000000000000000000000000000000..73a7188b56a4a43a99d6cab766098006d14ee878 Binary files /dev/null and b/data/train/face/5831.jpg differ diff --git a/data/train/face/6165.jpg b/data/train/face/6165.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5e7464501af714d64eb35e53aa526249e04781ee Binary files /dev/null and b/data/train/face/6165.jpg differ diff --git a/data/train/face/6380.jpg b/data/train/face/6380.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1a5d158cade027be396530ea67a4bc1c07662d9f Binary files /dev/null and b/data/train/face/6380.jpg differ diff --git a/data/train/face/6431.jpg b/data/train/face/6431.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6710ba0edfe88ead7118d01f435180c7c1e5a1e3 Binary files /dev/null and b/data/train/face/6431.jpg differ diff --git a/data/train/face/6440.jpg b/data/train/face/6440.jpg new file mode 100644 index 0000000000000000000000000000000000000000..edacc924b54a2c3b216eec1318cdf66f04f33644 Binary files /dev/null and b/data/train/face/6440.jpg differ diff --git a/data/train/face/6493.jpg b/data/train/face/6493.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e937536d6a3d9fa15c76cb365348469434babd0c Binary files /dev/null and b/data/train/face/6493.jpg differ diff --git a/data/train/face/6495.jpg b/data/train/face/6495.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6bf6cce641bece25e764d5cea10440bb9addb795 Binary files /dev/null and b/data/train/face/6495.jpg differ diff --git a/data/train/face/6542.jpg b/data/train/face/6542.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cc99fb4d4efe76d297e440559abca3fe05b7c659 Binary files /dev/null and b/data/train/face/6542.jpg differ diff --git a/data/train/face/6573.jpg b/data/train/face/6573.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2d704861dd8584efd69cc84b2f3e71467957cc14 Binary files /dev/null and b/data/train/face/6573.jpg differ diff --git a/data/train/face/6676.jpg b/data/train/face/6676.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c83e48b3e97d2880b8486d5a4b5b42b17331a4fb Binary files /dev/null and b/data/train/face/6676.jpg differ diff --git a/data/train/face/6781.jpg b/data/train/face/6781.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f3682ba8f4baacacc269d6d9ba089ee796e2473b Binary files /dev/null and b/data/train/face/6781.jpg differ diff --git a/data/train/face/6806.jpg b/data/train/face/6806.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2932960b2f4e5382065baec2747a1d23e6d4ffd1 Binary files /dev/null and b/data/train/face/6806.jpg differ diff --git a/data/train/face/6857.jpg b/data/train/face/6857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e74381a8da2e83ffedb0eab8ef605f19ebaf7c20 Binary files /dev/null and b/data/train/face/6857.jpg differ diff --git a/data/train/face/6920.jpg b/data/train/face/6920.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ead52df3c39c8629ef2b6e8f5c0f35f1f79b7d55 Binary files /dev/null and b/data/train/face/6920.jpg differ diff --git a/data/train/face/70.jpg b/data/train/face/70.jpg new file mode 100644 index 0000000000000000000000000000000000000000..832ef347ea4bd8d68d51271fbac5c41e529049bd Binary files /dev/null and b/data/train/face/70.jpg differ diff --git a/data/train/face/7128.jpg b/data/train/face/7128.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6df6177018d8ca1565cbb0fe5b870fe76a43d9c3 Binary files /dev/null and b/data/train/face/7128.jpg differ diff --git a/data/train/face/7209.jpg b/data/train/face/7209.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3ceabf21dda54ec60f4dde71afba00f00b4fe388 Binary files /dev/null and b/data/train/face/7209.jpg differ diff --git a/data/train/face/7299.jpg b/data/train/face/7299.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f1a33cd5b72cfba5646ccaa65c8e5219a6f1fc99 Binary files /dev/null and b/data/train/face/7299.jpg differ diff --git a/data/train/face/7321.jpg b/data/train/face/7321.jpg new file mode 100644 index 0000000000000000000000000000000000000000..adf82203989137bca2d5ec77f8b24accdac4526a Binary files /dev/null and b/data/train/face/7321.jpg differ diff --git a/data/train/face/7721.jpg b/data/train/face/7721.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1f900f0c9b6aac9e9b524511b012ff4d48f31d50 Binary files /dev/null and b/data/train/face/7721.jpg differ diff --git a/data/train/face/7823.jpg b/data/train/face/7823.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f9db6b78de95bb3dce88892d9591204729f80cda Binary files /dev/null and b/data/train/face/7823.jpg differ diff --git a/data/train/face/7853.jpg b/data/train/face/7853.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f0b5209981e3d63688f894e93930e8ade1769ba8 Binary files /dev/null and b/data/train/face/7853.jpg differ diff --git a/data/train/face/7930.jpg b/data/train/face/7930.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ddeba544e19e4a234df5b2908a6ad3c7e7e17164 Binary files /dev/null and b/data/train/face/7930.jpg differ diff --git a/data/train/face/7955.jpg b/data/train/face/7955.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2a159cb55f33bac8a6ea4612de4918801605d8fe Binary files /dev/null and b/data/train/face/7955.jpg differ diff --git a/data/train/face/8127.jpg b/data/train/face/8127.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c05825b3af61ceb722bce3da3cf858287d34b3a2 Binary files /dev/null and b/data/train/face/8127.jpg differ diff --git a/data/train/face/8174.jpg b/data/train/face/8174.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b830463fc0452d31a8934f129a4ccee5bc8873ed Binary files /dev/null and b/data/train/face/8174.jpg differ diff --git a/data/train/face/8251.jpg b/data/train/face/8251.jpg new file mode 100644 index 0000000000000000000000000000000000000000..99e5686c37f2ede6b592858495e5eb8929f9c016 Binary files /dev/null and b/data/train/face/8251.jpg differ diff --git a/data/train/face/8314.jpg b/data/train/face/8314.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7fbca481c0b44c6860c55cc35fd3ce37d49bc71b Binary files /dev/null and b/data/train/face/8314.jpg differ diff --git a/data/train/face/833.jpg b/data/train/face/833.jpg new file mode 100644 index 0000000000000000000000000000000000000000..25399b63437a0d10f845d948c79725efc8113904 Binary files /dev/null and b/data/train/face/833.jpg differ diff --git a/data/train/face/8332.jpg b/data/train/face/8332.jpg new file mode 100644 index 0000000000000000000000000000000000000000..96d6cb5d5d87ccfeedc5868e3b0288c2f1ab0ae2 Binary files /dev/null and b/data/train/face/8332.jpg differ diff --git a/data/train/face/8390.jpg b/data/train/face/8390.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5a5da0eeb69a94cc6e8a75c54243b3abb4c3cc39 Binary files /dev/null and b/data/train/face/8390.jpg differ diff --git a/data/train/face/8448.jpg b/data/train/face/8448.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ee6520951a456779c14ed3e8790f5c7a3669c7c1 Binary files /dev/null and b/data/train/face/8448.jpg differ diff --git a/data/train/face/8559.jpg b/data/train/face/8559.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7d1a183e2d8c16ddfcce32929f8d76a8dcd64828 Binary files /dev/null and b/data/train/face/8559.jpg differ diff --git a/data/train/face/8720.jpg b/data/train/face/8720.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c030151f9dcbf47861456c9929ab0e99bd100b45 Binary files /dev/null and b/data/train/face/8720.jpg differ diff --git a/data/train/face/8775.jpg b/data/train/face/8775.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cc20ec0a0ed1ee93620bfc82ded5e34cf8c87058 Binary files /dev/null and b/data/train/face/8775.jpg differ diff --git a/data/train/face/8834.jpg b/data/train/face/8834.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ad8630a59aae7c4782668436eee399d5cb3c2f2a Binary files /dev/null and b/data/train/face/8834.jpg differ diff --git a/data/train/face/8838.jpg b/data/train/face/8838.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e60107ec6fa16b45973698e7bbfa30f570d126aa Binary files /dev/null and b/data/train/face/8838.jpg differ diff --git a/data/train/face/8857.jpg b/data/train/face/8857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9c721e64b49d39fa3249e48a4ea50b963be86f25 Binary files /dev/null and b/data/train/face/8857.jpg differ diff --git a/data/train/face/8884.jpg b/data/train/face/8884.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c4449b3ac59e77fc47a42bed318372b87ddfab54 Binary files /dev/null and b/data/train/face/8884.jpg differ diff --git a/data/train/face/8910.jpg b/data/train/face/8910.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3a769828d0bf21eb279821dc61ebcebe5a0085bb Binary files /dev/null and b/data/train/face/8910.jpg differ diff --git a/data/train/face/8998.jpg b/data/train/face/8998.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4b8514b0170ecf0cdc9409b5433c1fa218bff586 Binary files /dev/null and b/data/train/face/8998.jpg differ diff --git a/data/train/face/9040.jpg b/data/train/face/9040.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b29c29fcd3c533427d8cede25abb6dddf18bb67f Binary files /dev/null and b/data/train/face/9040.jpg differ diff --git a/data/train/face/9125.jpg b/data/train/face/9125.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bac9384191fa8c643244d0635dac46d04139dba4 Binary files /dev/null and b/data/train/face/9125.jpg differ diff --git a/data/train/face/9160.jpg b/data/train/face/9160.jpg new file mode 100644 index 0000000000000000000000000000000000000000..0f5c1742bb538d1e8a12023fce0d63a08a0b51bc Binary files /dev/null and b/data/train/face/9160.jpg differ diff --git a/data/train/face/9173.jpg b/data/train/face/9173.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c29eeb25fbaf1650e28a68f919688dbe90dc73a4 Binary files /dev/null and b/data/train/face/9173.jpg differ diff --git a/data/train/face/9608.jpg b/data/train/face/9608.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8b52540e9983d59d65ff5a74ddf5acbbab19e0b5 Binary files /dev/null and b/data/train/face/9608.jpg differ diff --git a/data/train/face/9706.jpg b/data/train/face/9706.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e57a258f498e4af1566a6849c29310b10f875494 Binary files /dev/null and b/data/train/face/9706.jpg differ diff --git a/data/train/face/9723.jpg b/data/train/face/9723.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8bc921f845b77053051536895fe74e4b5c48310b Binary files /dev/null and b/data/train/face/9723.jpg differ diff --git a/data/train/face/9982.jpg b/data/train/face/9982.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ce0c8cf5e0435115ca86735d63f0fbd0efcf1842 Binary files /dev/null and b/data/train/face/9982.jpg differ diff --git a/data/train/face/9985.jpg b/data/train/face/9985.jpg new file mode 100644 index 0000000000000000000000000000000000000000..23e5a2543288845b2eb6c200deb99ed4a1f58575 Binary files /dev/null and b/data/train/face/9985.jpg differ diff --git a/data/val/comics/1002.jpg b/data/val/comics/1002.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5af06b5fc01b8a3ecc0c888b60367bb21ea5d347 Binary files /dev/null and b/data/val/comics/1002.jpg differ diff --git a/data/val/comics/1130.jpg b/data/val/comics/1130.jpg new file mode 100644 index 0000000000000000000000000000000000000000..0a140a2a5354d02ce7a16795554e64ccb769ae71 Binary files /dev/null and b/data/val/comics/1130.jpg differ diff --git a/data/val/comics/1189.jpg b/data/val/comics/1189.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ad2ebf1ccf22957f4355ce396e49101672c8b27b Binary files /dev/null and b/data/val/comics/1189.jpg differ diff --git a/data/val/comics/1218.jpg b/data/val/comics/1218.jpg new file mode 100644 index 0000000000000000000000000000000000000000..25ccb6825f0e06a62b61c7753c5e8f2fbb00ad6e Binary files /dev/null and b/data/val/comics/1218.jpg differ diff --git a/data/val/comics/1273.jpg b/data/val/comics/1273.jpg new file mode 100644 index 0000000000000000000000000000000000000000..36ea69a6ded180bc3663ce6eaa2773390e1c9fa3 Binary files /dev/null and b/data/val/comics/1273.jpg differ diff --git a/data/val/comics/1300.jpg b/data/val/comics/1300.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d529410d755f4a2da5eeeb0a8533126167b15d95 Binary files /dev/null and b/data/val/comics/1300.jpg differ diff --git a/data/val/comics/1339.jpg b/data/val/comics/1339.jpg new file mode 100644 index 0000000000000000000000000000000000000000..73a64c4b035972a50f3ff534230aae18ad58ca79 Binary files /dev/null and b/data/val/comics/1339.jpg differ diff --git a/data/val/comics/1370.jpg b/data/val/comics/1370.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8f7b3e8354e52145a6c3dd790fbf650839778600 Binary files /dev/null and b/data/val/comics/1370.jpg differ diff --git a/data/val/comics/1460.jpg b/data/val/comics/1460.jpg new file mode 100644 index 0000000000000000000000000000000000000000..351f8ba15086ba363dae995c2de84e9df6dc078f Binary files /dev/null and b/data/val/comics/1460.jpg differ diff --git a/data/val/comics/1471.jpg b/data/val/comics/1471.jpg new file mode 100644 index 0000000000000000000000000000000000000000..44e72114c95b663d45fe07ae77d0a67c4e1cd909 Binary files /dev/null and b/data/val/comics/1471.jpg differ diff --git a/data/val/comics/1479.jpg b/data/val/comics/1479.jpg new file mode 100644 index 0000000000000000000000000000000000000000..033b735418b646f06cbb1761e1d4eef2f9ce6a2e Binary files /dev/null and b/data/val/comics/1479.jpg differ diff --git a/data/val/comics/1804.jpg b/data/val/comics/1804.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8faf1c1af22fa4cdffe9fd2b983eee40d8a14417 Binary files /dev/null and b/data/val/comics/1804.jpg differ diff --git a/data/val/comics/1862.jpg b/data/val/comics/1862.jpg new file mode 100644 index 0000000000000000000000000000000000000000..760ab9a5da0f5bbfccae45339e0b7e5395fcaefe Binary files /dev/null and b/data/val/comics/1862.jpg differ diff --git a/data/val/comics/1927.jpg b/data/val/comics/1927.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6c81ae8a63fcea0c23c0af6e5850b05c179e4148 Binary files /dev/null and b/data/val/comics/1927.jpg differ diff --git a/data/val/comics/2078.jpg b/data/val/comics/2078.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d245fab0e5c3dcd899bb3b9904fec3885cb25af7 Binary files /dev/null and b/data/val/comics/2078.jpg differ diff --git a/data/val/comics/2101.jpg b/data/val/comics/2101.jpg new file mode 100644 index 0000000000000000000000000000000000000000..a9fe5232ae0ca4ef723d5be686b2261cd05f5ddc Binary files /dev/null and b/data/val/comics/2101.jpg differ diff --git a/data/val/comics/2131.jpg b/data/val/comics/2131.jpg new file mode 100644 index 0000000000000000000000000000000000000000..93db3e858ebcdf864e0269747f8a571b603860ee Binary files /dev/null and b/data/val/comics/2131.jpg differ diff --git a/data/val/comics/2231.jpg b/data/val/comics/2231.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5db9185a091cf01d2730034c8300596d8de3a358 Binary files /dev/null and b/data/val/comics/2231.jpg differ diff --git a/data/val/comics/2355.jpg b/data/val/comics/2355.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1f254fa1b1a8ed70c9584f1badf998af9da700df Binary files /dev/null and b/data/val/comics/2355.jpg differ diff --git a/data/val/comics/248.jpg b/data/val/comics/248.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7c0833de2ccd25233b2b261060c5156d5f9bc1dd Binary files /dev/null and b/data/val/comics/248.jpg differ diff --git a/data/val/comics/2508.jpg b/data/val/comics/2508.jpg new file mode 100644 index 0000000000000000000000000000000000000000..13b86497fd92191da4275dfa24578a805702b283 Binary files /dev/null and b/data/val/comics/2508.jpg differ diff --git a/data/val/comics/2514.jpg b/data/val/comics/2514.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2811521641222e270365f85bb8dffcf35af7e881 Binary files /dev/null and b/data/val/comics/2514.jpg differ diff --git a/data/val/comics/2772.jpg b/data/val/comics/2772.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3fad24ab3a7a09492a8587e4c12be8241d974c81 Binary files /dev/null and b/data/val/comics/2772.jpg differ diff --git a/data/val/comics/2857.jpg b/data/val/comics/2857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9649a69e8f8c9cfa887615f80f088383d0e379b8 Binary files /dev/null and b/data/val/comics/2857.jpg differ diff --git a/data/val/comics/312.jpg b/data/val/comics/312.jpg new file mode 100644 index 0000000000000000000000000000000000000000..52d8fb867eb9e7b4f73e1226ddcef00a5a9a7fab Binary files /dev/null and b/data/val/comics/312.jpg differ diff --git a/data/val/comics/3223.jpg b/data/val/comics/3223.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6bb1aebd5035d280ed6a753626e08597edf21b29 Binary files /dev/null and b/data/val/comics/3223.jpg differ diff --git a/data/val/comics/3322.jpg b/data/val/comics/3322.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1e72630d1f60de4c679d0317054ed4a6321ce182 Binary files /dev/null and b/data/val/comics/3322.jpg differ diff --git a/data/val/comics/3359.jpg b/data/val/comics/3359.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2fa644e58d070b6e19f12aa3fe6f2d56889b2dbd Binary files /dev/null and b/data/val/comics/3359.jpg differ diff --git a/data/val/comics/3428.jpg b/data/val/comics/3428.jpg new file mode 100644 index 0000000000000000000000000000000000000000..26a5beb6d41360b16ea3834c61fc1f4a71debf4a Binary files /dev/null and b/data/val/comics/3428.jpg differ diff --git a/data/val/comics/3520.jpg b/data/val/comics/3520.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d078a249d2dca1e285b5acf246c08c9821ac9655 Binary files /dev/null and b/data/val/comics/3520.jpg differ diff --git a/data/val/comics/3592.jpg b/data/val/comics/3592.jpg new file mode 100644 index 0000000000000000000000000000000000000000..522196b28d05107ab768c7d690f48b2408e829dd Binary files /dev/null and b/data/val/comics/3592.jpg differ diff --git a/data/val/comics/3733.jpg b/data/val/comics/3733.jpg new file mode 100644 index 0000000000000000000000000000000000000000..db991958f8d5549240799e584de47e6b3057acbd Binary files /dev/null and b/data/val/comics/3733.jpg differ diff --git a/data/val/comics/3876.jpg b/data/val/comics/3876.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3d7217d92c2ecc186214811e94230d97069a2f3c Binary files /dev/null and b/data/val/comics/3876.jpg differ diff --git a/data/val/comics/3970.jpg b/data/val/comics/3970.jpg new file mode 100644 index 0000000000000000000000000000000000000000..a816f4ddf35a28c5671627471c4f59fe3155a6a7 Binary files /dev/null and b/data/val/comics/3970.jpg differ diff --git a/data/val/comics/3978.jpg b/data/val/comics/3978.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2e91b0c7d4e4300289f73415aed4e09d1b08a42d Binary files /dev/null and b/data/val/comics/3978.jpg differ diff --git a/data/val/comics/40.jpg b/data/val/comics/40.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e2edd6de1def1c630dfb98204b7b9598bd8ed5ac Binary files /dev/null and b/data/val/comics/40.jpg differ diff --git a/data/val/comics/4075.jpg b/data/val/comics/4075.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7c0f6407f61cd6c1d514185917f487aa5b15ecc9 Binary files /dev/null and b/data/val/comics/4075.jpg differ diff --git a/data/val/comics/4081.jpg b/data/val/comics/4081.jpg new file mode 100644 index 0000000000000000000000000000000000000000..18121bab9426c48d8ca01bb791e7a672039eb44a Binary files /dev/null and b/data/val/comics/4081.jpg differ diff --git a/data/val/comics/416.jpg b/data/val/comics/416.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d10eafbf094c7e42a4d3d7aaafc2ee7cb3c53657 Binary files /dev/null and b/data/val/comics/416.jpg differ diff --git a/data/val/comics/4183.jpg b/data/val/comics/4183.jpg new file mode 100644 index 0000000000000000000000000000000000000000..eb52f0b93e40a5d22d98be81427c9b6051067fdd Binary files /dev/null and b/data/val/comics/4183.jpg differ diff --git a/data/val/comics/4368.jpg b/data/val/comics/4368.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f7aa7ea5d9b70185b7bc4b202d913f5439621f94 Binary files /dev/null and b/data/val/comics/4368.jpg differ diff --git a/data/val/comics/4613.jpg b/data/val/comics/4613.jpg new file mode 100644 index 0000000000000000000000000000000000000000..66f2a805b72abb7e0fd21e0afb9bf80bea804760 Binary files /dev/null and b/data/val/comics/4613.jpg differ diff --git a/data/val/comics/4660.jpg b/data/val/comics/4660.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ed4c6e913eecb8b862d81e035b8fe500d51e5910 Binary files /dev/null and b/data/val/comics/4660.jpg differ diff --git a/data/val/comics/4682.jpg b/data/val/comics/4682.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5a03eea276608d4d8b784e17ea17f1b9ea41321f Binary files /dev/null and b/data/val/comics/4682.jpg differ diff --git a/data/val/comics/4765.jpg b/data/val/comics/4765.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c3ffdd56a62f6dee785b92c96fa0fb1723a2e319 Binary files /dev/null and b/data/val/comics/4765.jpg differ diff --git a/data/val/comics/48.jpg b/data/val/comics/48.jpg new file mode 100644 index 0000000000000000000000000000000000000000..943da05be258d5444628834b96ec8352e457041e Binary files /dev/null and b/data/val/comics/48.jpg differ diff --git a/data/val/comics/5093.jpg b/data/val/comics/5093.jpg new file mode 100644 index 0000000000000000000000000000000000000000..14e83f51e440d74e231fd387b93233e09c46e57a Binary files /dev/null and b/data/val/comics/5093.jpg differ diff --git a/data/val/comics/5368.jpg b/data/val/comics/5368.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1b4634f2f9ab7012b8772361081765e1154b4dba Binary files /dev/null and b/data/val/comics/5368.jpg differ diff --git a/data/val/comics/5456.jpg b/data/val/comics/5456.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3e88276e647f3e6b3d263a9e08601bfbf994b074 Binary files /dev/null and b/data/val/comics/5456.jpg differ diff --git a/data/val/comics/5528.jpg b/data/val/comics/5528.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ffaa0f5cf2b5676813fde31414ff2c17b5495afd Binary files /dev/null and b/data/val/comics/5528.jpg differ diff --git a/data/val/comics/5831.jpg b/data/val/comics/5831.jpg new file mode 100644 index 0000000000000000000000000000000000000000..dfeb5cc7554e591efc523631a69d4ec474f78bee Binary files /dev/null and b/data/val/comics/5831.jpg differ diff --git a/data/val/comics/6165.jpg b/data/val/comics/6165.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d6900eab53c14a79c1554b7b8c14e5695ccfa078 Binary files /dev/null and b/data/val/comics/6165.jpg differ diff --git a/data/val/comics/6380.jpg b/data/val/comics/6380.jpg new file mode 100644 index 0000000000000000000000000000000000000000..681e8944c90ef84cfe3317a3279f6274e9216f54 Binary files /dev/null and b/data/val/comics/6380.jpg differ diff --git a/data/val/comics/6431.jpg b/data/val/comics/6431.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c97c7ce3bfcd2dc433eb5700f270eace0876bc2f Binary files /dev/null and b/data/val/comics/6431.jpg differ diff --git a/data/val/comics/6440.jpg b/data/val/comics/6440.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7fe37bb6d7bfb0237ae1a6686f157e17f48e577c Binary files /dev/null and b/data/val/comics/6440.jpg differ diff --git a/data/val/comics/6493.jpg b/data/val/comics/6493.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e3b85f4e7001df95e495c28446fb78a7d96ae5f9 Binary files /dev/null and b/data/val/comics/6493.jpg differ diff --git a/data/val/comics/6495.jpg b/data/val/comics/6495.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6fe581025aaa62079ef7bfcc2474123a38f04634 Binary files /dev/null and b/data/val/comics/6495.jpg differ diff --git a/data/val/comics/6542.jpg b/data/val/comics/6542.jpg new file mode 100644 index 0000000000000000000000000000000000000000..fdf8f9e8a4321a6d72b1f9759aa190c9b31248d7 Binary files /dev/null and b/data/val/comics/6542.jpg differ diff --git a/data/val/comics/6573.jpg b/data/val/comics/6573.jpg new file mode 100644 index 0000000000000000000000000000000000000000..74764d8f62a968ec4b1005c1983cca77669f5406 Binary files /dev/null and b/data/val/comics/6573.jpg differ diff --git a/data/val/comics/6676.jpg b/data/val/comics/6676.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e154fc985277a4713c241841510293cb118c390e Binary files /dev/null and b/data/val/comics/6676.jpg differ diff --git a/data/val/comics/6781.jpg b/data/val/comics/6781.jpg new file mode 100644 index 0000000000000000000000000000000000000000..74d13a38f830f21ea210a9bc21d0599911038923 Binary files /dev/null and b/data/val/comics/6781.jpg differ diff --git a/data/val/comics/6806.jpg b/data/val/comics/6806.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9e119ba5d365a7dd43789b28df73bdb9964c4d14 Binary files /dev/null and b/data/val/comics/6806.jpg differ diff --git a/data/val/comics/6857.jpg b/data/val/comics/6857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e2212d546fdaea4799d1064435535e42019d464c Binary files /dev/null and b/data/val/comics/6857.jpg differ diff --git a/data/val/comics/6920.jpg b/data/val/comics/6920.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7bea8b09420ccfa67ad70e5d25c6aecc85da7cf8 Binary files /dev/null and b/data/val/comics/6920.jpg differ diff --git a/data/val/comics/70.jpg b/data/val/comics/70.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5bd324d8a30b7503cfd6bc7da2421ca523945e86 Binary files /dev/null and b/data/val/comics/70.jpg differ diff --git a/data/val/comics/7128.jpg b/data/val/comics/7128.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ec78d95f4040fea214e4a3d6861beec70e26c3c0 Binary files /dev/null and b/data/val/comics/7128.jpg differ diff --git a/data/val/comics/7209.jpg b/data/val/comics/7209.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6301bbb8eba4d9e4742490c47aa6bc4a1cd45b38 Binary files /dev/null and b/data/val/comics/7209.jpg differ diff --git a/data/val/comics/7299.jpg b/data/val/comics/7299.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ff279f015b17d3c729fe7061f37f0393e62af1f5 Binary files /dev/null and b/data/val/comics/7299.jpg differ diff --git a/data/val/comics/7321.jpg b/data/val/comics/7321.jpg new file mode 100644 index 0000000000000000000000000000000000000000..87a634ce55090d697989f8c6ef6e85acf59a8f7d Binary files /dev/null and b/data/val/comics/7321.jpg differ diff --git a/data/val/comics/7721.jpg b/data/val/comics/7721.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c8df28c62a6c38cccffdb2347dec84ba360561ee Binary files /dev/null and b/data/val/comics/7721.jpg differ diff --git a/data/val/comics/7823.jpg b/data/val/comics/7823.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8302ecfb9a812bd683b40a181fc9266bfdf99464 Binary files /dev/null and b/data/val/comics/7823.jpg differ diff --git a/data/val/comics/7853.jpg b/data/val/comics/7853.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b67755b67d551d99028a5d5d7f1dd417f388ea68 Binary files /dev/null and b/data/val/comics/7853.jpg differ diff --git a/data/val/comics/7930.jpg b/data/val/comics/7930.jpg new file mode 100644 index 0000000000000000000000000000000000000000..870c5bb35e2c9b03eb4b1271d7e1aa74731bc966 Binary files /dev/null and b/data/val/comics/7930.jpg differ diff --git a/data/val/comics/7955.jpg b/data/val/comics/7955.jpg new file mode 100644 index 0000000000000000000000000000000000000000..610f889d59dca9df648cc3074a0b4b210b30a18d Binary files /dev/null and b/data/val/comics/7955.jpg differ diff --git a/data/val/comics/8127.jpg b/data/val/comics/8127.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cbc767ffec72d4aa8dd36685855f4e901ee9f73d Binary files /dev/null and b/data/val/comics/8127.jpg differ diff --git a/data/val/comics/8174.jpg b/data/val/comics/8174.jpg new file mode 100644 index 0000000000000000000000000000000000000000..19a0487ea9a6e657ae863e164cb78ea2b241dc42 Binary files /dev/null and b/data/val/comics/8174.jpg differ diff --git a/data/val/comics/8251.jpg b/data/val/comics/8251.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1c50c717654d23a55dc9bdfc7c5e0437940b9e6b Binary files /dev/null and b/data/val/comics/8251.jpg differ diff --git a/data/val/comics/8314.jpg b/data/val/comics/8314.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6e512a31513f006b5e3032ee99ebbc2c04c4fc83 Binary files /dev/null and b/data/val/comics/8314.jpg differ diff --git a/data/val/comics/833.jpg b/data/val/comics/833.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4d15b5b8d8a820fe2ebbda2def1a5a1eace54094 Binary files /dev/null and b/data/val/comics/833.jpg differ diff --git a/data/val/comics/8332.jpg b/data/val/comics/8332.jpg new file mode 100644 index 0000000000000000000000000000000000000000..765150e6404775f9cfba6ad001b26cee72927731 Binary files /dev/null and b/data/val/comics/8332.jpg differ diff --git a/data/val/comics/8390.jpg b/data/val/comics/8390.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8383cc8d91aa87731cc5318f7ea501bd71c4f84b Binary files /dev/null and b/data/val/comics/8390.jpg differ diff --git a/data/val/comics/8448.jpg b/data/val/comics/8448.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f4905e7f3afdaa9202c5efd143aeac5de483fe42 Binary files /dev/null and b/data/val/comics/8448.jpg differ diff --git a/data/val/comics/8559.jpg b/data/val/comics/8559.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ab2267f1bea0bbc4ed58716afa7c093d1cd504d9 Binary files /dev/null and b/data/val/comics/8559.jpg differ diff --git a/data/val/comics/8720.jpg b/data/val/comics/8720.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4b921e3b5091ac15b3a82b5d7d5b6606b75e0077 Binary files /dev/null and b/data/val/comics/8720.jpg differ diff --git a/data/val/comics/8775.jpg b/data/val/comics/8775.jpg new file mode 100644 index 0000000000000000000000000000000000000000..63af05582b2e75a9df05327ef86f53eb4a081ba0 Binary files /dev/null and b/data/val/comics/8775.jpg differ diff --git a/data/val/comics/8834.jpg b/data/val/comics/8834.jpg new file mode 100644 index 0000000000000000000000000000000000000000..0a54130d809de92bf29a951eadb509100cd96618 Binary files /dev/null and b/data/val/comics/8834.jpg differ diff --git a/data/val/comics/8838.jpg b/data/val/comics/8838.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ab2813c48f820068a058afd62869fdc8b152ec0c Binary files /dev/null and b/data/val/comics/8838.jpg differ diff --git a/data/val/comics/8857.jpg b/data/val/comics/8857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7b89a33f3feb4efbceaa53996bd22793f641eeb3 Binary files /dev/null and b/data/val/comics/8857.jpg differ diff --git a/data/val/comics/8884.jpg b/data/val/comics/8884.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2bd6603c71913a48f7facd998630c5bbafe46b0c Binary files /dev/null and b/data/val/comics/8884.jpg differ diff --git a/data/val/comics/8910.jpg b/data/val/comics/8910.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4bf4b2771cf0a941ea075b782f099d609d71be7d Binary files /dev/null and b/data/val/comics/8910.jpg differ diff --git a/data/val/comics/8998.jpg b/data/val/comics/8998.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bad044dfd36d146c2508371a0b2a2e35feaad64d Binary files /dev/null and b/data/val/comics/8998.jpg differ diff --git a/data/val/comics/9040.jpg b/data/val/comics/9040.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1c94a4e54f8834cad0099301e01c85e7e9f0a402 Binary files /dev/null and b/data/val/comics/9040.jpg differ diff --git a/data/val/comics/9125.jpg b/data/val/comics/9125.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ab66a89442330c3ef4fc7de66bb0d76bc7ea06cc Binary files /dev/null and b/data/val/comics/9125.jpg differ diff --git a/data/val/comics/9160.jpg b/data/val/comics/9160.jpg new file mode 100644 index 0000000000000000000000000000000000000000..04d77c79bc1c04dbb8d69aa3137abd37a180ccc4 Binary files /dev/null and b/data/val/comics/9160.jpg differ diff --git a/data/val/comics/9173.jpg b/data/val/comics/9173.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ddb0224e2f9e9e54abf5a0da31f3a57f9795b32e Binary files /dev/null and b/data/val/comics/9173.jpg differ diff --git a/data/val/comics/9608.jpg b/data/val/comics/9608.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ce2d0d77f4ad068901dcdc6e89288de26e560523 Binary files /dev/null and b/data/val/comics/9608.jpg differ diff --git a/data/val/comics/9706.jpg b/data/val/comics/9706.jpg new file mode 100644 index 0000000000000000000000000000000000000000..01e524794c7b60bb08960d1f3d82f7d00b6aa5fb Binary files /dev/null and b/data/val/comics/9706.jpg differ diff --git a/data/val/comics/9723.jpg b/data/val/comics/9723.jpg new file mode 100644 index 0000000000000000000000000000000000000000..24f563e02edec5aeba702da4482b60c19ffa62e7 Binary files /dev/null and b/data/val/comics/9723.jpg differ diff --git a/data/val/comics/9982.jpg b/data/val/comics/9982.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bfde43c299b88e210b6ea13ce6ed3f798ad89b39 Binary files /dev/null and b/data/val/comics/9982.jpg differ diff --git a/data/val/comics/9985.jpg b/data/val/comics/9985.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6788bc9ea81de75238b0917ef5a3dd5d1c7155d2 Binary files /dev/null and b/data/val/comics/9985.jpg differ diff --git a/data/val/face/1002.jpg b/data/val/face/1002.jpg new file mode 100644 index 0000000000000000000000000000000000000000..291f0be95d6276304c38369549ca4baae1436b5e Binary files /dev/null and b/data/val/face/1002.jpg differ diff --git a/data/val/face/1130.jpg b/data/val/face/1130.jpg new file mode 100644 index 0000000000000000000000000000000000000000..18b5a44e20167924653660f7092cb33e2178275e Binary files /dev/null and b/data/val/face/1130.jpg differ diff --git a/data/val/face/1189.jpg b/data/val/face/1189.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7ee438a634e86b2a5824bbdc1bc5a28044d706c9 Binary files /dev/null and b/data/val/face/1189.jpg differ diff --git a/data/val/face/1218.jpg b/data/val/face/1218.jpg new file mode 100644 index 0000000000000000000000000000000000000000..efda784ae31d92ee31fbc70489cc5b1c7890ef3f Binary files /dev/null and b/data/val/face/1218.jpg differ diff --git a/data/val/face/1273.jpg b/data/val/face/1273.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b50c2f5caaf04689d0f013556344edc1e0c43f2b Binary files /dev/null and b/data/val/face/1273.jpg differ diff --git a/data/val/face/1300.jpg b/data/val/face/1300.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7e78b364f76e8d90560de6ffacec4c309008be32 Binary files /dev/null and b/data/val/face/1300.jpg differ diff --git a/data/val/face/1339.jpg b/data/val/face/1339.jpg new file mode 100644 index 0000000000000000000000000000000000000000..629ec646c9e126f3fab4288e55a6ebc0e061cfdd Binary files /dev/null and b/data/val/face/1339.jpg differ diff --git a/data/val/face/1370.jpg b/data/val/face/1370.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ec4ba1d4320a153138e8bc59b08b3341f0fef5bf Binary files /dev/null and b/data/val/face/1370.jpg differ diff --git a/data/val/face/1460.jpg b/data/val/face/1460.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b2c9527925919e78cd5ca35dc12b291453cb0f4a Binary files /dev/null and b/data/val/face/1460.jpg differ diff --git a/data/val/face/1471.jpg b/data/val/face/1471.jpg new file mode 100644 index 0000000000000000000000000000000000000000..34fda2617a8eef410fdc3ea268d2f2f019a19919 Binary files /dev/null and b/data/val/face/1471.jpg differ diff --git a/data/val/face/1479.jpg b/data/val/face/1479.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d2a811bb8ace310e3d93da47a707b5489f38b1d6 Binary files /dev/null and b/data/val/face/1479.jpg differ diff --git a/data/val/face/1804.jpg b/data/val/face/1804.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8bdf311fbadb123c24990cd6f826c25f8ad260d5 Binary files /dev/null and b/data/val/face/1804.jpg differ diff --git a/data/val/face/1862.jpg b/data/val/face/1862.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bbe2920a23bcd2cc7d09bf0c7062883266921d16 Binary files /dev/null and b/data/val/face/1862.jpg differ diff --git a/data/val/face/1927.jpg b/data/val/face/1927.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2befded1d3aaa02e379d544fa9800460afd26420 Binary files /dev/null and b/data/val/face/1927.jpg differ diff --git a/data/val/face/2078.jpg b/data/val/face/2078.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c0f458b4715b8b0479d5773205a599fde82942ad Binary files /dev/null and b/data/val/face/2078.jpg differ diff --git a/data/val/face/2101.jpg b/data/val/face/2101.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7c586701cd1f4f6c8249f6b3b8fcb82904907e72 Binary files /dev/null and b/data/val/face/2101.jpg differ diff --git a/data/val/face/2131.jpg b/data/val/face/2131.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f68707bfb314f58ae0fdaffd260e68028651fc7c Binary files /dev/null and b/data/val/face/2131.jpg differ diff --git a/data/val/face/2231.jpg b/data/val/face/2231.jpg new file mode 100644 index 0000000000000000000000000000000000000000..79dbb92b89dab7448f8df454dd84b06c59967896 Binary files /dev/null and b/data/val/face/2231.jpg differ diff --git a/data/val/face/2355.jpg b/data/val/face/2355.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c69be7e90923ff6ad54f71253efcb00c5929e727 Binary files /dev/null and b/data/val/face/2355.jpg differ diff --git a/data/val/face/248.jpg b/data/val/face/248.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3393eb6f0b61017199238c9958067a45e4019505 Binary files /dev/null and b/data/val/face/248.jpg differ diff --git a/data/val/face/2508.jpg b/data/val/face/2508.jpg new file mode 100644 index 0000000000000000000000000000000000000000..dd42201550bbf9dff699064d16d3c7883c510216 Binary files /dev/null and b/data/val/face/2508.jpg differ diff --git a/data/val/face/2514.jpg b/data/val/face/2514.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4e94334cdffd46edc6ca87498c932a0d67f7e87f Binary files /dev/null and b/data/val/face/2514.jpg differ diff --git a/data/val/face/2772.jpg b/data/val/face/2772.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8078ef7ef26a18130b217a31dbdf51ab3378a766 Binary files /dev/null and b/data/val/face/2772.jpg differ diff --git a/data/val/face/2857.jpg b/data/val/face/2857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1d761d86a01f57c348536961ebf4cf099181e2f0 Binary files /dev/null and b/data/val/face/2857.jpg differ diff --git a/data/val/face/312.jpg b/data/val/face/312.jpg new file mode 100644 index 0000000000000000000000000000000000000000..395a7aaf6d7441935244e16dfc39c73393a07369 Binary files /dev/null and b/data/val/face/312.jpg differ diff --git a/data/val/face/3223.jpg b/data/val/face/3223.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1c0028c2cbd9496afc5e65daa08108db4376fa2a Binary files /dev/null and b/data/val/face/3223.jpg differ diff --git a/data/val/face/3322.jpg b/data/val/face/3322.jpg new file mode 100644 index 0000000000000000000000000000000000000000..24904ad6cea2d461f42218a065796540faf4db88 Binary files /dev/null and b/data/val/face/3322.jpg differ diff --git a/data/val/face/3359.jpg b/data/val/face/3359.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cae681ba3a5905f64f5f519a8585705920c35134 Binary files /dev/null and b/data/val/face/3359.jpg differ diff --git a/data/val/face/3428.jpg b/data/val/face/3428.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e345cf918b0ccd15ccc1d54fa207ef32f4e96880 Binary files /dev/null and b/data/val/face/3428.jpg differ diff --git a/data/val/face/3520.jpg b/data/val/face/3520.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bcfe5da0db98a30142c1cea5e42d04b9589058d2 Binary files /dev/null and b/data/val/face/3520.jpg differ diff --git a/data/val/face/3592.jpg b/data/val/face/3592.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f8be893cd9f45016738e863c26752c09c57b541e Binary files /dev/null and b/data/val/face/3592.jpg differ diff --git a/data/val/face/3733.jpg b/data/val/face/3733.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5db3fbfc9ec26d91a120c44d8b225ba2f7a9cba8 Binary files /dev/null and b/data/val/face/3733.jpg differ diff --git a/data/val/face/3876.jpg b/data/val/face/3876.jpg new file mode 100644 index 0000000000000000000000000000000000000000..99b9cf1ec6c0dc68f5bc433753838b0ca4465d95 Binary files /dev/null and b/data/val/face/3876.jpg differ diff --git a/data/val/face/3970.jpg b/data/val/face/3970.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6e10853032610193f9b7fe168e515069f434937d Binary files /dev/null and b/data/val/face/3970.jpg differ diff --git a/data/val/face/3978.jpg b/data/val/face/3978.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d9024676929e37d07fe2b0b0e1fef5f1eaf96a7c Binary files /dev/null and b/data/val/face/3978.jpg differ diff --git a/data/val/face/40.jpg b/data/val/face/40.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9156f0293d58a1ffe8f2cb1772698e324be3f269 Binary files /dev/null and b/data/val/face/40.jpg differ diff --git a/data/val/face/4075.jpg b/data/val/face/4075.jpg new file mode 100644 index 0000000000000000000000000000000000000000..988ff6923ae8ef94e681e6daab1f0493f0fbb2c5 Binary files /dev/null and b/data/val/face/4075.jpg differ diff --git a/data/val/face/4081.jpg b/data/val/face/4081.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7e6afa9a94a906616da6c34382f42376adde9c51 Binary files /dev/null and b/data/val/face/4081.jpg differ diff --git a/data/val/face/416.jpg b/data/val/face/416.jpg new file mode 100644 index 0000000000000000000000000000000000000000..92193c41e32776690dcc6b4056b216e2a70f67a8 Binary files /dev/null and b/data/val/face/416.jpg differ diff --git a/data/val/face/4183.jpg b/data/val/face/4183.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c0046c18640da00bdfbced94846fc48c7ff79090 Binary files /dev/null and b/data/val/face/4183.jpg differ diff --git a/data/val/face/4368.jpg b/data/val/face/4368.jpg new file mode 100644 index 0000000000000000000000000000000000000000..150a3e6c72ae83d9b9c54ccae2ba76a265238165 Binary files /dev/null and b/data/val/face/4368.jpg differ diff --git a/data/val/face/4613.jpg b/data/val/face/4613.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3891f226fbe506c3c08bef119f1a7ccd9d5f25c7 Binary files /dev/null and b/data/val/face/4613.jpg differ diff --git a/data/val/face/4660.jpg b/data/val/face/4660.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b72a1eddf3e6459c4a10939e05057e4677203989 Binary files /dev/null and b/data/val/face/4660.jpg differ diff --git a/data/val/face/4682.jpg b/data/val/face/4682.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e92d6d384b601e5747fa05b5294fb2e8fce247d9 Binary files /dev/null and b/data/val/face/4682.jpg differ diff --git a/data/val/face/4765.jpg b/data/val/face/4765.jpg new file mode 100644 index 0000000000000000000000000000000000000000..829cce02bee92a7b62446c32ae05930c9e9cb47c Binary files /dev/null and b/data/val/face/4765.jpg differ diff --git a/data/val/face/48.jpg b/data/val/face/48.jpg new file mode 100644 index 0000000000000000000000000000000000000000..595a7e42eaef212f8cfebf19417051fbddee30b2 Binary files /dev/null and b/data/val/face/48.jpg differ diff --git a/data/val/face/5093.jpg b/data/val/face/5093.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5dd387ce943ff232557d645cd8bf8002cd4fe243 Binary files /dev/null and b/data/val/face/5093.jpg differ diff --git a/data/val/face/5368.jpg b/data/val/face/5368.jpg new file mode 100644 index 0000000000000000000000000000000000000000..da8d5f88eed1a2166e650b0c9c21cc02c0b89c13 Binary files /dev/null and b/data/val/face/5368.jpg differ diff --git a/data/val/face/5456.jpg b/data/val/face/5456.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b1dc5061a98c0cd6bb705137f3a69427233f9850 Binary files /dev/null and b/data/val/face/5456.jpg differ diff --git a/data/val/face/5528.jpg b/data/val/face/5528.jpg new file mode 100644 index 0000000000000000000000000000000000000000..55456ec22a7686992014af348e4e70c1c7222765 Binary files /dev/null and b/data/val/face/5528.jpg differ diff --git a/data/val/face/5831.jpg b/data/val/face/5831.jpg new file mode 100644 index 0000000000000000000000000000000000000000..73a7188b56a4a43a99d6cab766098006d14ee878 Binary files /dev/null and b/data/val/face/5831.jpg differ diff --git a/data/val/face/6165.jpg b/data/val/face/6165.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5e7464501af714d64eb35e53aa526249e04781ee Binary files /dev/null and b/data/val/face/6165.jpg differ diff --git a/data/val/face/6380.jpg b/data/val/face/6380.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1a5d158cade027be396530ea67a4bc1c07662d9f Binary files /dev/null and b/data/val/face/6380.jpg differ diff --git a/data/val/face/6431.jpg b/data/val/face/6431.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6710ba0edfe88ead7118d01f435180c7c1e5a1e3 Binary files /dev/null and b/data/val/face/6431.jpg differ diff --git a/data/val/face/6440.jpg b/data/val/face/6440.jpg new file mode 100644 index 0000000000000000000000000000000000000000..edacc924b54a2c3b216eec1318cdf66f04f33644 Binary files /dev/null and b/data/val/face/6440.jpg differ diff --git a/data/val/face/6493.jpg b/data/val/face/6493.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e937536d6a3d9fa15c76cb365348469434babd0c Binary files /dev/null and b/data/val/face/6493.jpg differ diff --git a/data/val/face/6495.jpg b/data/val/face/6495.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6bf6cce641bece25e764d5cea10440bb9addb795 Binary files /dev/null and b/data/val/face/6495.jpg differ diff --git a/data/val/face/6542.jpg b/data/val/face/6542.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cc99fb4d4efe76d297e440559abca3fe05b7c659 Binary files /dev/null and b/data/val/face/6542.jpg differ diff --git a/data/val/face/6573.jpg b/data/val/face/6573.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2d704861dd8584efd69cc84b2f3e71467957cc14 Binary files /dev/null and b/data/val/face/6573.jpg differ diff --git a/data/val/face/6676.jpg b/data/val/face/6676.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c83e48b3e97d2880b8486d5a4b5b42b17331a4fb Binary files /dev/null and b/data/val/face/6676.jpg differ diff --git a/data/val/face/6781.jpg b/data/val/face/6781.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f3682ba8f4baacacc269d6d9ba089ee796e2473b Binary files /dev/null and b/data/val/face/6781.jpg differ diff --git a/data/val/face/6806.jpg b/data/val/face/6806.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2932960b2f4e5382065baec2747a1d23e6d4ffd1 Binary files /dev/null and b/data/val/face/6806.jpg differ diff --git a/data/val/face/6857.jpg b/data/val/face/6857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e74381a8da2e83ffedb0eab8ef605f19ebaf7c20 Binary files /dev/null and b/data/val/face/6857.jpg differ diff --git a/data/val/face/6920.jpg b/data/val/face/6920.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ead52df3c39c8629ef2b6e8f5c0f35f1f79b7d55 Binary files /dev/null and b/data/val/face/6920.jpg differ diff --git a/data/val/face/70.jpg b/data/val/face/70.jpg new file mode 100644 index 0000000000000000000000000000000000000000..832ef347ea4bd8d68d51271fbac5c41e529049bd Binary files /dev/null and b/data/val/face/70.jpg differ diff --git a/data/val/face/7128.jpg b/data/val/face/7128.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6df6177018d8ca1565cbb0fe5b870fe76a43d9c3 Binary files /dev/null and b/data/val/face/7128.jpg differ diff --git a/data/val/face/7209.jpg b/data/val/face/7209.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3ceabf21dda54ec60f4dde71afba00f00b4fe388 Binary files /dev/null and b/data/val/face/7209.jpg differ diff --git a/data/val/face/7299.jpg b/data/val/face/7299.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f1a33cd5b72cfba5646ccaa65c8e5219a6f1fc99 Binary files /dev/null and b/data/val/face/7299.jpg differ diff --git a/data/val/face/7321.jpg b/data/val/face/7321.jpg new file mode 100644 index 0000000000000000000000000000000000000000..adf82203989137bca2d5ec77f8b24accdac4526a Binary files /dev/null and b/data/val/face/7321.jpg differ diff --git a/data/val/face/7721.jpg b/data/val/face/7721.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1f900f0c9b6aac9e9b524511b012ff4d48f31d50 Binary files /dev/null and b/data/val/face/7721.jpg differ diff --git a/data/val/face/7823.jpg b/data/val/face/7823.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f9db6b78de95bb3dce88892d9591204729f80cda Binary files /dev/null and b/data/val/face/7823.jpg differ diff --git a/data/val/face/7853.jpg b/data/val/face/7853.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f0b5209981e3d63688f894e93930e8ade1769ba8 Binary files /dev/null and b/data/val/face/7853.jpg differ diff --git a/data/val/face/7930.jpg b/data/val/face/7930.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ddeba544e19e4a234df5b2908a6ad3c7e7e17164 Binary files /dev/null and b/data/val/face/7930.jpg differ diff --git a/data/val/face/7955.jpg b/data/val/face/7955.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2a159cb55f33bac8a6ea4612de4918801605d8fe Binary files /dev/null and b/data/val/face/7955.jpg differ diff --git a/data/val/face/8127.jpg b/data/val/face/8127.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c05825b3af61ceb722bce3da3cf858287d34b3a2 Binary files /dev/null and b/data/val/face/8127.jpg differ diff --git a/data/val/face/8174.jpg b/data/val/face/8174.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b830463fc0452d31a8934f129a4ccee5bc8873ed Binary files /dev/null and b/data/val/face/8174.jpg differ diff --git a/data/val/face/8251.jpg b/data/val/face/8251.jpg new file mode 100644 index 0000000000000000000000000000000000000000..99e5686c37f2ede6b592858495e5eb8929f9c016 Binary files /dev/null and b/data/val/face/8251.jpg differ diff --git a/data/val/face/8314.jpg b/data/val/face/8314.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7fbca481c0b44c6860c55cc35fd3ce37d49bc71b Binary files /dev/null and b/data/val/face/8314.jpg differ diff --git a/data/val/face/833.jpg b/data/val/face/833.jpg new file mode 100644 index 0000000000000000000000000000000000000000..25399b63437a0d10f845d948c79725efc8113904 Binary files /dev/null and b/data/val/face/833.jpg differ diff --git a/data/val/face/8332.jpg b/data/val/face/8332.jpg new file mode 100644 index 0000000000000000000000000000000000000000..96d6cb5d5d87ccfeedc5868e3b0288c2f1ab0ae2 Binary files /dev/null and b/data/val/face/8332.jpg differ diff --git a/data/val/face/8390.jpg b/data/val/face/8390.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5a5da0eeb69a94cc6e8a75c54243b3abb4c3cc39 Binary files /dev/null and b/data/val/face/8390.jpg differ diff --git a/data/val/face/8448.jpg b/data/val/face/8448.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ee6520951a456779c14ed3e8790f5c7a3669c7c1 Binary files /dev/null and b/data/val/face/8448.jpg differ diff --git a/data/val/face/8559.jpg b/data/val/face/8559.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7d1a183e2d8c16ddfcce32929f8d76a8dcd64828 Binary files /dev/null and b/data/val/face/8559.jpg differ diff --git a/data/val/face/8720.jpg b/data/val/face/8720.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c030151f9dcbf47861456c9929ab0e99bd100b45 Binary files /dev/null and b/data/val/face/8720.jpg differ diff --git a/data/val/face/8775.jpg b/data/val/face/8775.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cc20ec0a0ed1ee93620bfc82ded5e34cf8c87058 Binary files /dev/null and b/data/val/face/8775.jpg differ diff --git a/data/val/face/8834.jpg b/data/val/face/8834.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ad8630a59aae7c4782668436eee399d5cb3c2f2a Binary files /dev/null and b/data/val/face/8834.jpg differ diff --git a/data/val/face/8838.jpg b/data/val/face/8838.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e60107ec6fa16b45973698e7bbfa30f570d126aa Binary files /dev/null and b/data/val/face/8838.jpg differ diff --git a/data/val/face/8857.jpg b/data/val/face/8857.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9c721e64b49d39fa3249e48a4ea50b963be86f25 Binary files /dev/null and b/data/val/face/8857.jpg differ diff --git a/data/val/face/8884.jpg b/data/val/face/8884.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c4449b3ac59e77fc47a42bed318372b87ddfab54 Binary files /dev/null and b/data/val/face/8884.jpg differ diff --git a/data/val/face/8910.jpg b/data/val/face/8910.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3a769828d0bf21eb279821dc61ebcebe5a0085bb Binary files /dev/null and b/data/val/face/8910.jpg differ diff --git a/data/val/face/8998.jpg b/data/val/face/8998.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4b8514b0170ecf0cdc9409b5433c1fa218bff586 Binary files /dev/null and b/data/val/face/8998.jpg differ diff --git a/data/val/face/9040.jpg b/data/val/face/9040.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b29c29fcd3c533427d8cede25abb6dddf18bb67f Binary files /dev/null and b/data/val/face/9040.jpg differ diff --git a/data/val/face/9125.jpg b/data/val/face/9125.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bac9384191fa8c643244d0635dac46d04139dba4 Binary files /dev/null and b/data/val/face/9125.jpg differ diff --git a/data/val/face/9160.jpg b/data/val/face/9160.jpg new file mode 100644 index 0000000000000000000000000000000000000000..0f5c1742bb538d1e8a12023fce0d63a08a0b51bc Binary files /dev/null and b/data/val/face/9160.jpg differ diff --git a/data/val/face/9173.jpg b/data/val/face/9173.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c29eeb25fbaf1650e28a68f919688dbe90dc73a4 Binary files /dev/null and b/data/val/face/9173.jpg differ diff --git a/data/val/face/9608.jpg b/data/val/face/9608.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8b52540e9983d59d65ff5a74ddf5acbbab19e0b5 Binary files /dev/null and b/data/val/face/9608.jpg differ diff --git a/data/val/face/9706.jpg b/data/val/face/9706.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e57a258f498e4af1566a6849c29310b10f875494 Binary files /dev/null and b/data/val/face/9706.jpg differ diff --git a/data/val/face/9723.jpg b/data/val/face/9723.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8bc921f845b77053051536895fe74e4b5c48310b Binary files /dev/null and b/data/val/face/9723.jpg differ diff --git a/data/val/face/9982.jpg b/data/val/face/9982.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ce0c8cf5e0435115ca86735d63f0fbd0efcf1842 Binary files /dev/null and b/data/val/face/9982.jpg differ diff --git a/data/val/face/9985.jpg b/data/val/face/9985.jpg new file mode 100644 index 0000000000000000000000000000000000000000..23e5a2543288845b2eb6c200deb99ed4a1f58575 Binary files /dev/null and b/data/val/face/9985.jpg differ diff --git a/dataset.py b/dataset.py new file mode 100644 index 0000000000000000000000000000000000000000..b61fa5524da52fef0c675376e5a7da1c7a41f9f9 --- /dev/null +++ b/dataset.py @@ -0,0 +1,48 @@ +import numpy as np +import config +import os +from PIL import Image +from torch.utils.data import Dataset, DataLoader +from torchvision.utils import save_image + + +class MapDataset(Dataset): + def __init__(self, root_dir_comics, root_dir_face): + self.root_dir_face = root_dir_face + self.list_files_face = os.listdir(self.root_dir_face) + self.root_dir_comics = root_dir_comics + self.list_files_comics = os.listdir(self.root_dir_comics) + + def __len__(self): + return len(self.list_files_face) + + def __getitem__(self, index): + img_file_input = self.list_files_face[index] + img_file_target = self.list_files_comics[index] + img_path_input = os.path.join(self.root_dir_face, img_file_input) + img_path_target = os.path.join(self.root_dir_comics, img_file_target) + image_input = np.array(Image.open(img_path_input)) + image_target = np.array(Image.open(img_path_target)) + input_image = image_input + target_image = image_target + + augmentations = config.both_transform(image=input_image, image0=target_image) + input_image = augmentations["image"] + target_image = augmentations["image0"] + + input_image = config.transform_only_input(image=input_image)["image"] + target_image = config.transform_only_mask(image=target_image)["image"] + + return input_image, target_image + + +if __name__ == "__main__": + dataset = MapDataset("data/train/") + loader = DataLoader(dataset, batch_size=5) + for x, y in loader: + print(x.shape) + save_image(x, "x.png") + save_image(y, "y.png") + import sys + + sys.exit() \ No newline at end of file diff --git a/disc.pth.tar b/disc.pth.tar new file mode 100644 index 0000000000000000000000000000000000000000..206341aa9b3bfd7145754c15c17a0aa4473e21e6 --- /dev/null +++ b/disc.pth.tar @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b9eade8fe1c334aedd6df0ec1906fc98988e370ab2ba0c1e0de78e3f85bb99d9 +size 33250221 diff --git a/discriminator_model.py b/discriminator_model.py new file mode 100644 index 0000000000000000000000000000000000000000..bcae8e3d6588d0623af80889d1a78b6f8685af38 --- /dev/null +++ b/discriminator_model.py @@ -0,0 +1,68 @@ +import torch +import torch.nn as nn + + +class CNNBlock(nn.Module): + def __init__(self, in_channels, out_channels, stride): + super(CNNBlock, self).__init__() + self.conv = nn.Sequential( + nn.Conv2d( + in_channels, out_channels, 4, stride, 1, bias=False, padding_mode="reflect" + ), + nn.BatchNorm2d(out_channels), + nn.LeakyReLU(0.2), + ) + + def forward(self, x): + return self.conv(x) + + +class Discriminator(nn.Module): + def __init__(self, in_channels=3, features=[64, 128, 256, 512]): + super().__init__() + self.initial = nn.Sequential( + nn.Conv2d( + in_channels * 2, + features[0], + kernel_size=4, + stride=2, + padding=1, + padding_mode="reflect", + ), + nn.LeakyReLU(0.2), + ) + + layers = [] + in_channels = features[0] + for feature in features[1:]: + layers.append( + CNNBlock(in_channels, feature, stride=1 if feature == features[-1] else 2), + ) + in_channels = feature + + layers.append( + nn.Conv2d( + in_channels, 1, kernel_size=4, stride=1, padding=1, padding_mode="reflect" + ), + ) + + self.model = nn.Sequential(*layers) + + def forward(self, x, y): + x = torch.cat([x, y], dim=1) + x = self.initial(x) + x = self.model(x) + return x + + +def test(): + x = torch.randn((1, 3, 256, 256)) + y = torch.randn((1, 3, 256, 256)) + model = Discriminator(in_channels=3) + preds = model(x, y) + print(model) + print(preds.shape) + + +if __name__ == "__main__": + test() \ No newline at end of file diff --git a/evaluation/input_0.png b/evaluation/input_0.png new file mode 100644 index 0000000000000000000000000000000000000000..ed2460dd561be7cc9acba02f5573abcbb53be2f3 Binary files /dev/null and b/evaluation/input_0.png differ diff --git a/evaluation/y_gen_0.png b/evaluation/y_gen_0.png new file mode 100644 index 0000000000000000000000000000000000000000..0464d2ce23b415953e9578ea676df4b9d3fd846b Binary files /dev/null and b/evaluation/y_gen_0.png differ diff --git a/gen.pth.tar b/gen.pth.tar new file mode 100644 index 0000000000000000000000000000000000000000..ac28053731ee02bad64ce7f52cf21eefc5801e9a --- /dev/null +++ b/gen.pth.tar @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ebe48025f3276240c856d829928655efd1c171945ea4782240a0c6a4051e39e1 +size 653080267 diff --git a/generator_model.py b/generator_model.py new file mode 100644 index 0000000000000000000000000000000000000000..826e0b09c150204cbffc863057f594d44038e2ed --- /dev/null +++ b/generator_model.py @@ -0,0 +1,101 @@ +import torch +import torch.nn as nn + + +class Block(nn.Module): + def __init__(self, in_channels, out_channels, down=True, act="relu", use_dropout=False): + super(Block, self).__init__() + self.conv = nn.Sequential( + nn.Conv2d(in_channels, out_channels, 4, 2, 1, bias=False, padding_mode="reflect") + if down + else nn.ConvTranspose2d(in_channels, out_channels, 4, 2, 1, bias=False), + nn.BatchNorm2d(out_channels), + nn.ReLU() if act == "relu" else nn.LeakyReLU(0.2), + ) + + self.use_dropout = use_dropout + self.dropout = nn.Dropout(0.5) + self.down = down + + def forward(self, x): + x = self.conv(x) + return self.dropout(x) if self.use_dropout else x + + +class Generator(nn.Module): + def __init__(self, in_channels=3, features=64): + super().__init__() + self.initial_down = nn.Sequential( + nn.Conv2d(in_channels, features, 4, 2, 1, padding_mode="reflect"), + nn.LeakyReLU(0.2), + ) + self.down1 = Block(features, features * 2, down=True, act="leaky", use_dropout=False) + self.down2 = Block( + features * 2, features * 4, down=True, act="leaky", use_dropout=False + ) + self.down3 = Block( + features * 4, features * 8, down=True, act="leaky", use_dropout=False + ) + self.down4 = Block( + features * 8, features * 8, down=True, act="leaky", use_dropout=False + ) + self.down5 = Block( + features * 8, features * 8, down=True, act="leaky", use_dropout=False + ) + self.down6 = Block( + features * 8, features * 8, down=True, act="leaky", use_dropout=False + ) + self.bottleneck = nn.Sequential( + nn.Conv2d(features * 8, features * 8, 4, 2, 1), nn.ReLU() + ) + + self.up1 = Block(features * 8, features * 8, down=False, act="relu", use_dropout=True) + self.up2 = Block( + features * 8 * 2, features * 8, down=False, act="relu", use_dropout=True + ) + self.up3 = Block( + features * 8 * 2, features * 8, down=False, act="relu", use_dropout=True + ) + self.up4 = Block( + features * 8 * 2, features * 8, down=False, act="relu", use_dropout=False + ) + self.up5 = Block( + features * 8 * 2, features * 4, down=False, act="relu", use_dropout=False + ) + self.up6 = Block( + features * 4 * 2, features * 2, down=False, act="relu", use_dropout=False + ) + self.up7 = Block(features * 2 * 2, features, down=False, act="relu", use_dropout=False) + self.final_up = nn.Sequential( + nn.ConvTranspose2d(features * 2, in_channels, kernel_size=4, stride=2, padding=1), + nn.Tanh(), + ) + + def forward(self, x): + d1 = self.initial_down(x) + d2 = self.down1(d1) + d3 = self.down2(d2) + d4 = self.down3(d3) + d5 = self.down4(d4) + d6 = self.down5(d5) + d7 = self.down6(d6) + bottleneck = self.bottleneck(d7) + up1 = self.up1(bottleneck) + up2 = self.up2(torch.cat([up1, d7], 1)) + up3 = self.up3(torch.cat([up2, d6], 1)) + up4 = self.up4(torch.cat([up3, d5], 1)) + up5 = self.up5(torch.cat([up4, d4], 1)) + up6 = self.up6(torch.cat([up5, d3], 1)) + up7 = self.up7(torch.cat([up6, d2], 1)) + return self.final_up(torch.cat([up7, d1], 1)) + + +def test(): + x = torch.randn((1, 3, 256, 256)) + model = Generator(in_channels=3, features=64) + preds = model(x) + print(preds.shape) + + +if __name__ == "__main__": + test() \ No newline at end of file diff --git a/input/WhatsApp Image 2023-08-06 at 5.14.34 PM.jpeg b/input/WhatsApp Image 2023-08-06 at 5.14.34 PM.jpeg new file mode 100644 index 0000000000000000000000000000000000000000..3ce7609fe12cda4fa6cd34df7976dc4b90da037a Binary files /dev/null and b/input/WhatsApp Image 2023-08-06 at 5.14.34 PM.jpeg differ diff --git a/input/comics/WhatsApp Image 2023-08-06 at 5.14.34 PM.jpeg b/input/comics/WhatsApp Image 2023-08-06 at 5.14.34 PM.jpeg new file mode 100644 index 0000000000000000000000000000000000000000..3ce7609fe12cda4fa6cd34df7976dc4b90da037a Binary files /dev/null and b/input/comics/WhatsApp Image 2023-08-06 at 5.14.34 PM.jpeg differ diff --git a/input/face/3189.jpg b/input/face/3189.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ba370115639a9844bef8a42f86ac9068622a520f Binary files /dev/null and b/input/face/3189.jpg differ diff --git a/train.py b/train.py new file mode 100644 index 0000000000000000000000000000000000000000..d61a72b3ebde698065b4024b2036e048abbe8198 --- /dev/null +++ b/train.py @@ -0,0 +1,102 @@ +import torch +from utils import save_checkpoint, load_checkpoint, save_some_examples +import torch.nn as nn +import torch.optim as optim +import config +from dataset import MapDataset +from generator_model import Generator +from discriminator_model import Discriminator +from torch.utils.data import DataLoader +from tqdm import tqdm +from torchvision.utils import save_image + +torch.backends.cudnn.benchmark = True + + +def train_fn( + disc, gen, loader, opt_disc, opt_gen, l1_loss, bce, g_scaler, d_scaler, +): + loop = tqdm(loader, leave=True) + + for idx, (x, y) in enumerate(loop): + x = x.to(config.DEVICE) + y = y.to(config.DEVICE) + + # Train Discriminator + with torch.cuda.amp.autocast(): + y_fake = gen(x) + D_real = disc(x, y) + D_real_loss = bce(D_real, torch.ones_like(D_real)) + D_fake = disc(x, y_fake.detach()) + D_fake_loss = bce(D_fake, torch.zeros_like(D_fake)) + D_loss = (D_real_loss + D_fake_loss) / 2 + + disc.zero_grad() + d_scaler.scale(D_loss).backward() + d_scaler.step(opt_disc) + d_scaler.update() + + # Train generator + with torch.cuda.amp.autocast(): + D_fake = disc(x, y_fake) + G_fake_loss = bce(D_fake, torch.ones_like(D_fake)) + L1 = l1_loss(y_fake, y) * config.L1_LAMBDA + G_loss = G_fake_loss + L1 + + opt_gen.zero_grad() + g_scaler.scale(G_loss).backward() + g_scaler.step(opt_gen) + g_scaler.update() + + if idx % 10 == 0: + loop.set_postfix( + D_real=torch.sigmoid(D_real).mean().item(), + D_fake=torch.sigmoid(D_fake).mean().item(), + ) + + +def main(): + disc = Discriminator(in_channels=3).to(config.DEVICE) + gen = Generator(in_channels=3, features=64).to(config.DEVICE) + opt_disc = optim.Adam(disc.parameters(), lr=config.LEARNING_RATE, betas=(0.5, 0.999), ) + opt_gen = optim.Adam(gen.parameters(), lr=config.LEARNING_RATE, betas=(0.5, 0.999)) + BCE = nn.BCEWithLogitsLoss() + L1_LOSS = nn.L1Loss() + + if config.LOAD_MODEL: + load_checkpoint( + config.CHECKPOINT_GEN, gen, opt_gen, config.LEARNING_RATE, + ) + load_checkpoint( + config.CHECKPOINT_DISC, disc, opt_disc, config.LEARNING_RATE, + ) + + train_dataset = MapDataset(root_dir_face=config.TRAIN_DIR + "/face", root_dir_comics=config.TRAIN_DIR + "/comics") + train_loader = DataLoader( + train_dataset, + batch_size=config.BATCH_SIZE, + shuffle=True, + num_workers=config.NUM_WORKERS, + ) + g_scaler = torch.cuda.amp.GradScaler() + d_scaler = torch.cuda.amp.GradScaler() + val_dataset = MapDataset(root_dir_face=config.VAL_DIR + "/face", root_dir_comics=config.VAL_DIR + "/comics") + val_loader = DataLoader(val_dataset, batch_size=4, shuffle=False) + + input_data = MapDataset(root_dir_face="input/face", root_dir_comics="input/comics") + input_loader = DataLoader(input_data, batch_size=1, shuffle=False) + + for epoch in range(config.NUM_EPOCHS): + # train_fn( + # disc, gen, train_loader, opt_disc, opt_gen, L1_LOSS, BCE, g_scaler, d_scaler, + # ) + # + # if config.SAVE_MODEL and epoch % 5 == 0: + # save_checkpoint(gen, opt_gen, filename=config.CHECKPOINT_GEN) + # save_checkpoint(disc, opt_disc, filename=config.CHECKPOINT_DISC) + + save_some_examples(gen, input_loader, epoch, folder="evaluation") + + +if __name__ == "__main__": + main() diff --git a/utils.py b/utils.py new file mode 100644 index 0000000000000000000000000000000000000000..46ecbd7096477ebad80306358e9ebc1e1234bcd9 --- /dev/null +++ b/utils.py @@ -0,0 +1,38 @@ +import torch +import config +from torchvision.utils import save_image + + +def save_some_examples(gen, val_loader, epoch, folder): + x, y = next(iter(val_loader)) + x, y = x.to(config.DEVICE), y.to(config.DEVICE) + gen.eval() + with torch.no_grad(): + y_fake = gen(x) + y_fake = y_fake * 0.5 + 0.5 # remove normalization# + save_image(y_fake, folder + f"/y_gen_{epoch}.png") + save_image(x * 0.5 + 0.5, folder + f"/input_{epoch}.png") + if epoch == 1: + save_image(y * 0.5 + 0.5, folder + f"/label_{epoch}.png") + gen.train() + + +def save_checkpoint(model, optimizer, filename="my_checkpoint.pth.tar"): + print("=> Saving checkpoint") + checkpoint = { + "state_dict": model.state_dict(), + "optimizer": optimizer.state_dict(), + } + torch.save(checkpoint, filename) + + +def load_checkpoint(checkpoint_file, model, optimizer, lr): + print("=> Loading checkpoint") + checkpoint = torch.load(checkpoint_file, map_location=config.DEVICE) + model.load_state_dict(checkpoint["state_dict"]) + optimizer.load_state_dict(checkpoint["optimizer"]) + + # If we don't do this then it will just have learning rate of old checkpoint + # and it will lead to many hours of debugging \: + for param_group in optimizer.param_groups: + param_group["lr"] = lr