From 3865e956256d8be9c0013d5609af57b0afec9f2e Mon Sep 17 00:00:00 2001 From: farnaz-frd <53252795+farnaz-frd@users.noreply.github.com> Date: Thu, 31 Mar 2022 22:38:47 +0430 Subject: [PATCH] Update image.py in line 10 --> ground_truth to ground-truth --- image.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/image.py b/image.py index c8506d49..72a5cc44 100644 --- a/image.py +++ b/image.py @@ -7,7 +7,7 @@ import cv2 def load_data(img_path,train = True): - gt_path = img_path.replace('.jpg','.h5').replace('images','ground_truth') + gt_path = img_path.replace('.jpg','.h5').replace('images','ground-truth') img = Image.open(img_path).convert('RGB') gt_file = h5py.File(gt_path) target = np.asarray(gt_file['density']) @@ -40,4 +40,4 @@ def load_data(img_path,train = True): target = cv2.resize(target,(target.shape[1]/8,target.shape[0]/8),interpolation = cv2.INTER_CUBIC)*64 - return img,target \ No newline at end of file + return img,target