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

big bug... not unique imaged :D

Browse Source
This commit is contained in:
Till Raab 2023-10-26 08:56:43 +02:00
parent efd860a127
commit 9905b6039e
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
datasets.py
View File

@ -24,6 +24,9 @@ class CustomDataset(Dataset):
self.dir_path = dir_path self.dir_path = dir_path
self.bbox_df = bbox_df self.bbox_df = bbox_df
embed()
quit()
self.all_images = np.array(sorted(self.bbox_df['image']), dtype=str) 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() # embed()

1
inference.py
View File

@ -63,7 +63,6 @@ if __name__ == '__main__':
test_data = create_train_or_test_dataset(TRAIN_DIR, train=False) test_data = create_train_or_test_dataset(TRAIN_DIR, train=False)
test_loader = create_valid_loader(test_data) test_loader = create_valid_loader(test_data)
infere_model(test_loader, model) infere_model(test_loader, model)
# detection_threshold = 0.8 # detection_threshold = 0.8