raab/efishSignalDetector
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

bug fixed ... no double images any longer... repeat trainging because there we have the same issue...

Browse Source
This commit is contained in:
Till Raab 2023-10-26 09:00:38 +02:00
parent 9905b6039e
commit 4373b7c453
1 changed files with 1 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
datasets.py
View File

@ -24,20 +24,8 @@ class CustomDataset(Dataset):
self.dir_path = dir_path self.dir_path = dir_path
self.bbox_df = bbox_df self.bbox_df = bbox_df
embed() self.all_images = np.array(sorted(pd.unique(self.bbox_df['image'])), dtype=str)
quit()
self.all_images = np.array(sorted(self.bbox_df['image']), dtype=str)
self.image_paths = list(map(lambda x: Path(self.dir_path)/x, self.all_images)) self.image_paths = list(map(lambda x: Path(self.dir_path)/x, self.all_images))
# embed()
# quit()
# self.image_paths = glob.glob(f'{self.dir_path}/*.png')
# self.all_images = [img_path.split(os.path.sep)[-1] for img_path in self.image_paths]
# self.all_images = np.array(sorted(self.all_images), dtype=str)
# if hasattr(use_idxs, '__len__'):
# self.all_images = self.all_images[use_idxs]
# self.bbox_df = pd.read_csv(os.path.join(dir_path, 'bbox_dataset.csv'), sep=',', index_col=0)
def __getitem__(self, idx): def __getitem__(self, idx):
image_name = self.all_images[idx] image_name = self.all_images[idx]