recoilme commited on
Commit
7e2c607
·
1 Parent(s): 19de470
dataset.py CHANGED
@@ -18,17 +18,17 @@ from tqdm import tqdm
18
  from datetime import timedelta
19
 
20
  # ---------------- 1️⃣ Настройки ----------------
21
- dtype = torch.float16
22
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
23
- batch_size = 10
24
  min_size = 384 #320 #192 #256 #192
25
  max_size = 768 #640 #384 #256 #384
26
  step = 32 #64
27
  empty_share = 0.0
28
  limit = 0
29
  # Основная процедура обработки
30
- folder_path = "/workspace/dataset/dataset/ae3" #alchemist"
31
- save_path = "/workspace/ae3_768" #"alchemist"
32
  os.makedirs(save_path, exist_ok=True)
33
 
34
  # Функция для очистки CUDA памяти
 
18
  from datetime import timedelta
19
 
20
  # ---------------- 1️⃣ Настройки ----------------
21
+ dtype = torch.float32
22
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
23
+ batch_size = 5
24
  min_size = 384 #320 #192 #256 #192
25
  max_size = 768 #640 #384 #256 #384
26
  step = 32 #64
27
  empty_share = 0.0
28
  limit = 0
29
  # Основная процедура обработки
30
+ folder_path = "/workspace/ds" #alchemist"
31
+ save_path = "/workspace/768" #"alchemist"
32
  os.makedirs(save_path, exist_ok=True)
33
 
34
  # Функция для очистки CUDA памяти
samples/unet_320x640_0.jpg CHANGED

Git LFS Details

  • SHA256: cbd28220db18f95d0fee2e027af5bc8829b9d3ee788c1d94b521d5c37c368983
  • Pointer size: 130 Bytes
  • Size of remote file: 62.4 kB

Git LFS Details

  • SHA256: e06f7cb07d9fbc3ef1d02347b7081a0135288569f11afa406ab7f62147972ce9
  • Pointer size: 130 Bytes
  • Size of remote file: 57.7 kB
samples/unet_384x640_0.jpg CHANGED

Git LFS Details

  • SHA256: c63700ae77198f9ffe16431218661d6703daaf085b5ebc5b9f5591b582d16bfd
  • Pointer size: 131 Bytes
  • Size of remote file: 159 kB

Git LFS Details

  • SHA256: 94a19d8a77792e26e560be6f197f73f480a5c7ffc838064aa17ecd96cb3e6d9e
  • Pointer size: 131 Bytes
  • Size of remote file: 147 kB
samples/unet_448x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 465c5bf3e1b611dd046d0d28b1bc4850c2562fef837bf85816ae00959f28f78c
  • Pointer size: 130 Bytes
  • Size of remote file: 81.7 kB

Git LFS Details

  • SHA256: 4fd8d002614216d688ce0174780d95a6523dc3f2f5e1223579327f6a867af1da
  • Pointer size: 130 Bytes
  • Size of remote file: 61.1 kB
samples/unet_512x640_0.jpg CHANGED

Git LFS Details

  • SHA256: e94978b4e71b7b45166215ce68d794a95dcd5ad4f3ea39f31be1189110fd958f
  • Pointer size: 131 Bytes
  • Size of remote file: 131 kB

Git LFS Details

  • SHA256: 7599a443c04103aa0f12f66b025c60f87840c3dd7210a73691d9bd6021b32ebd
  • Pointer size: 131 Bytes
  • Size of remote file: 120 kB
samples/unet_576x640_0.jpg CHANGED

Git LFS Details

  • SHA256: e3cd197d0f08c8da4e5e3c51258c4dae62ee498dfef2d876d9c0e93a7a2d8fc4
  • Pointer size: 131 Bytes
  • Size of remote file: 244 kB

Git LFS Details

  • SHA256: 072e19364e211e8873f625398b3914a071947c3762a3626452f3206f2aeec010
  • Pointer size: 131 Bytes
  • Size of remote file: 221 kB
samples/unet_640x320_0.jpg CHANGED

Git LFS Details

  • SHA256: bbaebc70f65baa27de7ef43a6028c9bf11a73d7cf2874e402183edc590006ee8
  • Pointer size: 130 Bytes
  • Size of remote file: 88.3 kB

Git LFS Details

  • SHA256: c0c88e2f8d6939470dc3ae885311fe3ae757e2f0faddefe3804c67049fe69a7b
  • Pointer size: 130 Bytes
  • Size of remote file: 92.8 kB
samples/unet_640x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 31315d6c63ecadd9ba9a9d568ae00047c2236a8d9cf87e9e357e477db0e427fc
  • Pointer size: 131 Bytes
  • Size of remote file: 110 kB

Git LFS Details

  • SHA256: fd3842f9b7102e36692abdd74ef26e101f5f678ece49a43bf6496bb2355a5326
  • Pointer size: 131 Bytes
  • Size of remote file: 117 kB
samples/unet_640x448_0.jpg CHANGED

Git LFS Details

  • SHA256: b98d7ce8c61c9670407c5480614e695a8ccfee23c37fd08297edc6e8b4f59355
  • Pointer size: 130 Bytes
  • Size of remote file: 92.2 kB

Git LFS Details

  • SHA256: 336b9180e730bf5c636f1c1fe5895101b095c22535fc3bf69ba9cf8dc692f840
  • Pointer size: 130 Bytes
  • Size of remote file: 95.2 kB
samples/unet_640x512_0.jpg CHANGED

Git LFS Details

  • SHA256: 5f29af11e61ea1773a84ab8cc9fa880bb60c1fdef4584be9157e9f1f8adb3d19
  • Pointer size: 131 Bytes
  • Size of remote file: 122 kB

Git LFS Details

  • SHA256: 7574b2431018192cecdc7d930ae6611f0354bb772fee34f563b0d31f5d1f770a
  • Pointer size: 130 Bytes
  • Size of remote file: 92.7 kB
samples/unet_640x576_0.jpg CHANGED

Git LFS Details

  • SHA256: e4263e6d85ffbbf5b0fd0cb3545ffb9fb9a081e73098e714e853f196a000845e
  • Pointer size: 131 Bytes
  • Size of remote file: 182 kB

Git LFS Details

  • SHA256: 8621a659518cb09f7c5b4480a22ff5f6d374d933e4f78fbc311dd0731248df60
  • Pointer size: 131 Bytes
  • Size of remote file: 167 kB
samples/unet_640x640_0.jpg CHANGED

Git LFS Details

  • SHA256: c839b3deba43cf54c33c903c7df431797f17c5734cf766ae3c0c32d84a3a6847
  • Pointer size: 131 Bytes
  • Size of remote file: 202 kB

Git LFS Details

  • SHA256: ec3f3a6b678b7389ea73639fb66f6d4908c9b07c23568d0caa272fbd6f6a076a
  • Pointer size: 131 Bytes
  • Size of remote file: 182 kB
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:08b55f95640f0615bc957b7e0641973220578146f32d1647f900fa74c93f1f4d
3
  size 6205958296
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e34aafa7f71c3293eefa66c3532db6f461403ee852c5870e8406cbf076ed051d
3
  size 6205958296