From 22efba57a351489242d4b6d37a4a3c6f0a5f6b28 Mon Sep 17 00:00:00 2001 From: iperov Date: Sat, 11 May 2019 19:44:01 +0400 Subject: [PATCH] fix --- samplelib/SampleProcessor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index a40679f..0921e52 100644 --- a/samplelib/SampleProcessor.py +++ b/samplelib/SampleProcessor.py @@ -212,8 +212,8 @@ class SampleProcessor(object): img_bgr = img[...,0:3] img_mask = img[...,3:4] - if apply_ct: - if ct_sample_bgr is not None: + if apply_ct and ct_sample is not None: + if ct_sample_bgr is None: ct_sample_bgr = ct_sample.load_bgr() ct_sample_bgr_resized = cv2.resize( ct_sample_bgr, (resolution,resolution), cv2.INTER_LINEAR )