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