WIP
parent
254a9e9561
commit
c11c3f7a48
@ -0,0 +1,31 @@
|
|||||||
|
from longinus.wikimedia.loaders.pytorch import WikimediaCommonsLoader
|
||||||
|
from longinus.utils.image import Image
|
||||||
|
|
||||||
|
class DatasetLoader():
|
||||||
|
def __init__(self, dataset_size):
|
||||||
|
self.dataset = list()
|
||||||
|
self.dataset_size = dataset_size
|
||||||
|
tmp_dataset = WikimediaCommonsLoader(batch_size=32, resize_to=512, crop_to=512)
|
||||||
|
counter = 0
|
||||||
|
for image in tmp_dataset:
|
||||||
|
self.dataset.append(image)
|
||||||
|
counter+=1
|
||||||
|
if counter >= self.dataset_size:
|
||||||
|
break
|
||||||
|
|
||||||
|
|
||||||
|
def get_dataset(self):
|
||||||
|
return self.dataset
|
||||||
|
|
||||||
|
def get_dataset_size(self):
|
||||||
|
return self.dataset_size
|
||||||
|
|
||||||
|
class Dataset():
|
||||||
|
def __init__(self, dataset_loader: DatasetLoader, dataset_size: int):
|
||||||
|
self.dataset = dataset_loader(dataset_size).get_dataset()
|
||||||
|
self.dataset_converted = self.convert_images()
|
||||||
|
|
||||||
|
def convert_images(self) -> list[Image]:
|
||||||
|
for image in self.dataset:
|
||||||
|
new_image = Image.load_image_from_mem(image)
|
||||||
|
pass
|
@ -0,0 +1 @@
|
|||||||
|
Subproject commit 39fa5b4ee0ff12a7c9c36b656b389f3e3a6fddef
|
Loading…
Reference in New Issue