From bde69ee2057a3c60794594cc4f0a8828bcee1e33 Mon Sep 17 00:00:00 2001 From: jubjamie Date: Thu, 23 Feb 2017 14:33:49 +0000 Subject: [PATCH] Update cv2 BGR2RGB Caller --- library/localization.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/library/localization.py b/library/localization.py index ff3f96f..9168432 100644 --- a/library/localization.py +++ b/library/localization.py @@ -94,7 +94,7 @@ def visualize(im, bbox_cords): mask[ymin:ymax,xmin:xmax] = 1 new_im = new_im*mask new_im = new_im.astype(np.uint8) - new_im = cv2.cvtColor(new_im, cv2.cv.CV_BGR2RGB) + new_im = cv2.cvtColor(new_im, cv2.COLOR_BGR2RGB) plt.imshow(new_im) def calculate_area(c): @@ -153,4 +153,4 @@ def main(): bbox_coords = bbox(im,model,layer,n_neurons,kmax) #Visualize the Box on the original image visualize(im, bbox_cords) - return bbox_cords \ No newline at end of file + return bbox_cords