From 9e7921c5cd78411f5823d3e53e0e461fe6ec71a5 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Mon, 12 Feb 2024 10:39:28 +0100
Subject: [PATCH] [test] Remove check of square image
Samples are not square cropped during data loading anymore due to recent
changes. Cropping is model-dependent.
---
tests/conftest.py | 11 -----------
1 file changed, 11 deletions(-)
diff --git a/tests/conftest.py b/tests/conftest.py
index 40276ff1..7b4fe9e5 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -160,19 +160,9 @@ class DatabaseCheckers:
Parameters
----------
-<<<<<<< HEAD
split
An instance of DatabaseSplit.
lengths
-=======
-
- make_split
- A database specific function that takes a split name and returns
- the loaded database split.
- split_filename
- This is the split we will check.
- lenghts
->>>>>>> 91bcad6 ([test] Add checks for specific image shapes)
A dictionary that contains keys matching those of the split (this will
be checked). The values of the dictionary should correspond to the
sizes of each of the datasets in the split.
@@ -236,7 +226,6 @@ class DatabaseCheckers:
assert isinstance(batch[0], torch.Tensor)
assert batch[0].shape[0] == batch_size # mini-batch size
assert batch[0].shape[1] == color_planes
- assert batch[0].shape[2] == batch[0].shape[3] # image is square
if expected_image_shape:
assert all(
--
GitLab