25 lines
774 B
Python
25 lines
774 B
Python
# script to test data loader by itself
|
|
# run from training root, edit the data_root manually
|
|
# python ldm/data/test_dl.py
|
|
from ldm.data.image_train_item import ImageTrainItem
|
|
import glob
|
|
import os
|
|
|
|
data_root = "training_samples\multiaspect"
|
|
|
|
for idx, f in enumerate(glob.iglob(f"{data_root}/*.jpg")):
|
|
for i in range(0, 40):
|
|
#print(f)
|
|
#image: PIL.Image, caption: str, target_wh: list, pathname: str, flip_p=0.0):
|
|
caption = os.path.basename(f)
|
|
caption = os.path.splitext(caption)[0]
|
|
my_iti = ImageTrainItem(None,caption,[512,512],f,0.0)
|
|
|
|
my_iti = my_iti.hydrate()
|
|
|
|
out_file_path = os.path.join(data_root, "output", f"{caption}_{i}.jpg")
|
|
#print(out_file_path)
|
|
my_iti.cropped_img.save(out_file_path)
|
|
|
|
|