Spaces:
Build error
Build error
Yamato commited on
Commit ·
bd88c34
1
Parent(s): b4d95df
Upload 209 files
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- .gitattributes +78 -0
- __pycache__/config.cpython-311.pyc +0 -0
- __pycache__/dataset.cpython-311.pyc +0 -0
- __pycache__/discriminator_model.cpython-311.pyc +0 -0
- __pycache__/generator_model.cpython-311.pyc +0 -0
- __pycache__/utils.cpython-311.pyc +0 -0
- config.py +39 -0
- data/train/059D7AA.jpg +0 -0
- data/train/08644B5.jpg +3 -0
- data/train/1631821863.jpg +0 -0
- data/train/1639101717_1-pornotaran-com-p-porno-sisi-nyu-v-odezhde-i-bez-1.jpg +0 -0
- data/train/1641166941_3-titis-org-p-with-and-without-clothes-erotika-4.jpg +0 -0
- data/train/1641166946_32-titis-org-p-with-and-without-clothes-erotika-36.jpg +0 -0
- data/train/1641166994_12-titis-org-p-with-and-without-clothes-erotika-14.jpg +0 -0
- data/train/1641167013_56-titis-org-p-with-and-without-clothes-erotika-63.jpg +0 -0
- data/train/1641167023_15-titis-org-p-with-and-without-clothes-erotika-17.jpg +0 -0
- data/train/1641167035_24-titis-org-p-with-and-without-clothes-erotika-28.jpg +0 -0
- data/train/1F7700C.jpg +0 -0
- data/train/205638-what-lies-beneath.jpg +0 -0
- data/train/257288-onoff-wife-likes-group-sex.jpg +0 -0
- data/train/294DF81.jpg +0 -0
- data/train/355878.jpg +0 -0
- data/train/35BAF4A.jpg +0 -0
- data/train/368373.jpg +0 -0
- data/train/3CC356F.jpg +0 -0
- data/train/4183F37.jpg +0 -0
- data/train/425010.jpg +0 -0
- data/train/524684-on-off-gym-clothes-edition_880x660.jpg +0 -0
- data/train/6E5FEC5.jpg +0 -0
- data/train/F442F6D.jpg +3 -0
- data/train/F4454CF.jpg +0 -0
- data/train/FCE968A.jpg +0 -0
- data/train/IMG_20230622_004242.jpg +3 -0
- data/train/RDT_20230816_2223042539738600720277542.jpg +0 -0
- data/train/RDT_20230816_2223425793397015014508204.jpg +0 -0
- data/train/RDT_20230816_2224063053527716817291251.jpg +0 -0
- data/train/RDT_20230816_2224113299449225084525931.jpg +0 -0
- data/train/RDT_20230816_2224154321247418314056723.jpg +0 -0
- data/train/RDT_20230816_2224393151687206565079569.jpg +0 -0
- data/train/RDT_20230816_2225527843510578657362280.jpg +0 -0
- data/train/RDT_20230816_2226026010272585724780483.jpg +0 -0
- data/train/RDT_20230816_2226187600371162520645577.jpg +0 -0
- data/train/RDT_20230816_2226246453321377833203060.jpg +0 -0
- data/train/RDT_20230816_2226505968921622465924570.jpg +0 -0
- data/train/RDT_20230816_2227027924590143163305337.jpg +0 -0
- data/train/RDT_20230816_2227133247335695971953400.jpg +0 -0
- data/train/RDT_20230816_2230274877066279641604337.jpg +0 -0
- data/train/RDT_20230816_2230444197156933897032462.jpg +0 -0
- data/train/RDT_20230816_223052214063475630422981.jpg +0 -0
- data/train/RDT_20230816_2231084093312944250189736.jpg +0 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,81 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
data/train/08644B5.jpg filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
data/train/F442F6D.jpg filter=lfs diff=lfs merge=lfs -text
|
| 38 |
+
data/train/IMG_20230622_004242.jpg filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
data/train/output10.jpg filter=lfs diff=lfs merge=lfs -text
|
| 40 |
+
data/train/output11.jpg filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
data/train/output12.jpg filter=lfs diff=lfs merge=lfs -text
|
| 42 |
+
data/train/output13.jpg filter=lfs diff=lfs merge=lfs -text
|
| 43 |
+
data/train/output14.jpg filter=lfs diff=lfs merge=lfs -text
|
| 44 |
+
data/train/output15.jpg filter=lfs diff=lfs merge=lfs -text
|
| 45 |
+
data/train/output16.jpg filter=lfs diff=lfs merge=lfs -text
|
| 46 |
+
data/train/output17.jpg filter=lfs diff=lfs merge=lfs -text
|
| 47 |
+
data/train/output18.jpg filter=lfs diff=lfs merge=lfs -text
|
| 48 |
+
data/train/output19.jpg filter=lfs diff=lfs merge=lfs -text
|
| 49 |
+
data/train/output2.jpg filter=lfs diff=lfs merge=lfs -text
|
| 50 |
+
data/train/output20.jpg filter=lfs diff=lfs merge=lfs -text
|
| 51 |
+
data/train/output21.jpg filter=lfs diff=lfs merge=lfs -text
|
| 52 |
+
data/train/output22.jpg filter=lfs diff=lfs merge=lfs -text
|
| 53 |
+
data/train/output23.jpg filter=lfs diff=lfs merge=lfs -text
|
| 54 |
+
data/train/output24.jpg filter=lfs diff=lfs merge=lfs -text
|
| 55 |
+
data/train/output25.jpg filter=lfs diff=lfs merge=lfs -text
|
| 56 |
+
data/train/output26.jpg filter=lfs diff=lfs merge=lfs -text
|
| 57 |
+
data/train/output27.jpg filter=lfs diff=lfs merge=lfs -text
|
| 58 |
+
data/train/output28.jpg filter=lfs diff=lfs merge=lfs -text
|
| 59 |
+
data/train/output29.jpg filter=lfs diff=lfs merge=lfs -text
|
| 60 |
+
data/train/output3.jpg filter=lfs diff=lfs merge=lfs -text
|
| 61 |
+
data/train/output30.jpg filter=lfs diff=lfs merge=lfs -text
|
| 62 |
+
data/train/output31.jpg filter=lfs diff=lfs merge=lfs -text
|
| 63 |
+
data/train/output32.jpg filter=lfs diff=lfs merge=lfs -text
|
| 64 |
+
data/train/output33.jpg filter=lfs diff=lfs merge=lfs -text
|
| 65 |
+
data/train/output34.jpg filter=lfs diff=lfs merge=lfs -text
|
| 66 |
+
data/train/output35.jpg filter=lfs diff=lfs merge=lfs -text
|
| 67 |
+
data/train/output36.jpg filter=lfs diff=lfs merge=lfs -text
|
| 68 |
+
data/train/output37.jpg filter=lfs diff=lfs merge=lfs -text
|
| 69 |
+
data/train/output4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 70 |
+
data/train/output5.jpg filter=lfs diff=lfs merge=lfs -text
|
| 71 |
+
data/train/output6.jpg filter=lfs diff=lfs merge=lfs -text
|
| 72 |
+
data/train/output7.jpg filter=lfs diff=lfs merge=lfs -text
|
| 73 |
+
data/train/output8.jpg filter=lfs diff=lfs merge=lfs -text
|
| 74 |
+
data/train/output9.jpg filter=lfs diff=lfs merge=lfs -text
|
| 75 |
+
data/val/08644B5.jpg filter=lfs diff=lfs merge=lfs -text
|
| 76 |
+
data/val/F442F6D.jpg filter=lfs diff=lfs merge=lfs -text
|
| 77 |
+
data/val/IMG_20230622_004242.jpg filter=lfs diff=lfs merge=lfs -text
|
| 78 |
+
data/val/output10.jpg filter=lfs diff=lfs merge=lfs -text
|
| 79 |
+
data/val/output11.jpg filter=lfs diff=lfs merge=lfs -text
|
| 80 |
+
data/val/output12.jpg filter=lfs diff=lfs merge=lfs -text
|
| 81 |
+
data/val/output13.jpg filter=lfs diff=lfs merge=lfs -text
|
| 82 |
+
data/val/output14.jpg filter=lfs diff=lfs merge=lfs -text
|
| 83 |
+
data/val/output15.jpg filter=lfs diff=lfs merge=lfs -text
|
| 84 |
+
data/val/output16.jpg filter=lfs diff=lfs merge=lfs -text
|
| 85 |
+
data/val/output17.jpg filter=lfs diff=lfs merge=lfs -text
|
| 86 |
+
data/val/output18.jpg filter=lfs diff=lfs merge=lfs -text
|
| 87 |
+
data/val/output19.jpg filter=lfs diff=lfs merge=lfs -text
|
| 88 |
+
data/val/output2.jpg filter=lfs diff=lfs merge=lfs -text
|
| 89 |
+
data/val/output20.jpg filter=lfs diff=lfs merge=lfs -text
|
| 90 |
+
data/val/output21.jpg filter=lfs diff=lfs merge=lfs -text
|
| 91 |
+
data/val/output22.jpg filter=lfs diff=lfs merge=lfs -text
|
| 92 |
+
data/val/output23.jpg filter=lfs diff=lfs merge=lfs -text
|
| 93 |
+
data/val/output24.jpg filter=lfs diff=lfs merge=lfs -text
|
| 94 |
+
data/val/output25.jpg filter=lfs diff=lfs merge=lfs -text
|
| 95 |
+
data/val/output26.jpg filter=lfs diff=lfs merge=lfs -text
|
| 96 |
+
data/val/output27.jpg filter=lfs diff=lfs merge=lfs -text
|
| 97 |
+
data/val/output28.jpg filter=lfs diff=lfs merge=lfs -text
|
| 98 |
+
data/val/output29.jpg filter=lfs diff=lfs merge=lfs -text
|
| 99 |
+
data/val/output3.jpg filter=lfs diff=lfs merge=lfs -text
|
| 100 |
+
data/val/output30.jpg filter=lfs diff=lfs merge=lfs -text
|
| 101 |
+
data/val/output31.jpg filter=lfs diff=lfs merge=lfs -text
|
| 102 |
+
data/val/output32.jpg filter=lfs diff=lfs merge=lfs -text
|
| 103 |
+
data/val/output33.jpg filter=lfs diff=lfs merge=lfs -text
|
| 104 |
+
data/val/output34.jpg filter=lfs diff=lfs merge=lfs -text
|
| 105 |
+
data/val/output35.jpg filter=lfs diff=lfs merge=lfs -text
|
| 106 |
+
data/val/output36.jpg filter=lfs diff=lfs merge=lfs -text
|
| 107 |
+
data/val/output37.jpg filter=lfs diff=lfs merge=lfs -text
|
| 108 |
+
data/val/output4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 109 |
+
data/val/output5.jpg filter=lfs diff=lfs merge=lfs -text
|
| 110 |
+
data/val/output6.jpg filter=lfs diff=lfs merge=lfs -text
|
| 111 |
+
data/val/output7.jpg filter=lfs diff=lfs merge=lfs -text
|
| 112 |
+
data/val/output8.jpg filter=lfs diff=lfs merge=lfs -text
|
| 113 |
+
data/val/output9.jpg filter=lfs diff=lfs merge=lfs -text
|
__pycache__/config.cpython-311.pyc
ADDED
|
Binary file (1.54 kB). View file
|
|
|
__pycache__/dataset.cpython-311.pyc
ADDED
|
Binary file (2.51 kB). View file
|
|
|
__pycache__/discriminator_model.cpython-311.pyc
ADDED
|
Binary file (3.71 kB). View file
|
|
|
__pycache__/generator_model.cpython-311.pyc
ADDED
|
Binary file (6.75 kB). View file
|
|
|
__pycache__/utils.cpython-311.pyc
ADDED
|
Binary file (2.64 kB). View file
|
|
|
config.py
ADDED
|
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import torch
|
| 2 |
+
import albumentations as A
|
| 3 |
+
from albumentations.pytorch import ToTensorV2
|
| 4 |
+
|
| 5 |
+
DEVICE = "cpu"
|
| 6 |
+
TRAIN_DIR = "data/train"
|
| 7 |
+
VAL_DIR = "data/val"
|
| 8 |
+
LEARNING_RATE = 2e-4
|
| 9 |
+
BATCH_SIZE = 16
|
| 10 |
+
NUM_WORKERS = 2
|
| 11 |
+
IMAGE_SIZE = 256
|
| 12 |
+
CHANNELS_IMG = 3
|
| 13 |
+
L1_LAMBDA = 100
|
| 14 |
+
LAMBDA_GP = 10
|
| 15 |
+
NUM_EPOCHS = 1
|
| 16 |
+
LOAD_MODEL = True
|
| 17 |
+
SAVE_MODEL = False
|
| 18 |
+
CHECKPOINT_DISC = "disc.pth.tar"
|
| 19 |
+
CHECKPOINT_GEN = "gen.pth.tar"
|
| 20 |
+
|
| 21 |
+
both_transform = A.Compose(
|
| 22 |
+
[A.Resize(width=256, height=256), ], additional_targets={"image0": "image"}, is_check_shapes=False
|
| 23 |
+
)
|
| 24 |
+
|
| 25 |
+
transform_only_input = A.Compose(
|
| 26 |
+
[
|
| 27 |
+
A.HorizontalFlip(p=0.5),
|
| 28 |
+
A.ColorJitter(p=0.2),
|
| 29 |
+
A.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5], max_pixel_value=255.0, ),
|
| 30 |
+
ToTensorV2(),
|
| 31 |
+
]
|
| 32 |
+
)
|
| 33 |
+
|
| 34 |
+
transform_only_mask = A.Compose(
|
| 35 |
+
[
|
| 36 |
+
A.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5], max_pixel_value=255.0, ),
|
| 37 |
+
ToTensorV2(),
|
| 38 |
+
]
|
| 39 |
+
)
|
data/train/059D7AA.jpg
ADDED
|
data/train/08644B5.jpg
ADDED
|
Git LFS Details
|
data/train/1631821863.jpg
ADDED
|
data/train/1639101717_1-pornotaran-com-p-porno-sisi-nyu-v-odezhde-i-bez-1.jpg
ADDED
|
data/train/1641166941_3-titis-org-p-with-and-without-clothes-erotika-4.jpg
ADDED
|
data/train/1641166946_32-titis-org-p-with-and-without-clothes-erotika-36.jpg
ADDED
|
data/train/1641166994_12-titis-org-p-with-and-without-clothes-erotika-14.jpg
ADDED
|
data/train/1641167013_56-titis-org-p-with-and-without-clothes-erotika-63.jpg
ADDED
|
data/train/1641167023_15-titis-org-p-with-and-without-clothes-erotika-17.jpg
ADDED
|
data/train/1641167035_24-titis-org-p-with-and-without-clothes-erotika-28.jpg
ADDED
|
data/train/1F7700C.jpg
ADDED
|
data/train/205638-what-lies-beneath.jpg
ADDED
|
data/train/257288-onoff-wife-likes-group-sex.jpg
ADDED
|
data/train/294DF81.jpg
ADDED
|
data/train/355878.jpg
ADDED
|
data/train/35BAF4A.jpg
ADDED
|
data/train/368373.jpg
ADDED
|
data/train/3CC356F.jpg
ADDED
|
data/train/4183F37.jpg
ADDED
|
data/train/425010.jpg
ADDED
|
data/train/524684-on-off-gym-clothes-edition_880x660.jpg
ADDED
|
data/train/6E5FEC5.jpg
ADDED
|
data/train/F442F6D.jpg
ADDED
|
Git LFS Details
|
data/train/F4454CF.jpg
ADDED
|
data/train/FCE968A.jpg
ADDED
|
data/train/IMG_20230622_004242.jpg
ADDED
|
Git LFS Details
|
data/train/RDT_20230816_2223042539738600720277542.jpg
ADDED
|
data/train/RDT_20230816_2223425793397015014508204.jpg
ADDED
|
data/train/RDT_20230816_2224063053527716817291251.jpg
ADDED
|
data/train/RDT_20230816_2224113299449225084525931.jpg
ADDED
|
data/train/RDT_20230816_2224154321247418314056723.jpg
ADDED
|
data/train/RDT_20230816_2224393151687206565079569.jpg
ADDED
|
data/train/RDT_20230816_2225527843510578657362280.jpg
ADDED
|
data/train/RDT_20230816_2226026010272585724780483.jpg
ADDED
|
data/train/RDT_20230816_2226187600371162520645577.jpg
ADDED
|
data/train/RDT_20230816_2226246453321377833203060.jpg
ADDED
|
data/train/RDT_20230816_2226505968921622465924570.jpg
ADDED
|
data/train/RDT_20230816_2227027924590143163305337.jpg
ADDED
|
data/train/RDT_20230816_2227133247335695971953400.jpg
ADDED
|
data/train/RDT_20230816_2230274877066279641604337.jpg
ADDED
|
data/train/RDT_20230816_2230444197156933897032462.jpg
ADDED
|
data/train/RDT_20230816_223052214063475630422981.jpg
ADDED
|
data/train/RDT_20230816_2231084093312944250189736.jpg
ADDED
|