big bug... not unique imaged :D

2023-10-26 08:56:43 +02:00 · 2023-10-26 08:56:43 +02:00 · 9905b6039e
commit 9905b6039e
parent efd860a127
2 changed files with 3 additions and 1 deletions
--- a/datasets.py
+++ b/datasets.py
@ -24,6 +24,9 @@ class CustomDataset(Dataset):
        self.dir_path = dir_path
        self.bbox_df = bbox_df

+        embed()
+        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))
        # embed()
--- a/inference.py
+++ b/inference.py
@ -63,7 +63,6 @@ if __name__ == '__main__':
    test_data = create_train_or_test_dataset(TRAIN_DIR, train=False)
    test_loader = create_valid_loader(test_data)

-
    infere_model(test_loader, model)

    # detection_threshold = 0.8