s495728/Detection-of-plant-diseases
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feature/load-dataset #2

Merged
s495733 merged 7 commits from feature/load-dataset into main 2024-05-05 19:42:12 +02:00
Conversation 0 Commits 7 Files Changed 6 +1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit c6f6ae28ca - Show all commits

1
dataset/dataset.py
View File

@ -34,6 +34,7 @@ class Dataset:
self.dataset = self.__load_dataset()\ self.dataset = self.__load_dataset()\
.shuffle(self.shuffle_buffer_size, seed=self.seed)\ .shuffle(self.shuffle_buffer_size, seed=self.seed)\
.repeat(self.repeat)\ .repeat(self.repeat)\
.batch(self.batch_size, drop_remainder=True)\
.prefetch(tf.data.experimental.AUTOTUNE) .prefetch(tf.data.experimental.AUTOTUNE)
def __load_dataset(self) -> tf.data.Dataset: def __load_dataset(self) -> tf.data.Dataset: