From afc93cf498fdff4cd8529bf54757367fbec1811e Mon Sep 17 00:00:00 2001 From: Yorusora Date: Thu, 29 May 2025 21:21:02 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=BA=86=E4=B8=AD=E4=BD=93?= =?UTF-8?q?=E5=8A=A8=E6=98=BE=E7=A4=BA=E5=BC=82=E5=B8=B8=E7=9A=84=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- func/Module_bcg_quality_label.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/func/Module_bcg_quality_label.py b/func/Module_bcg_quality_label.py index b524db7..d903e9f 100644 --- a/func/Module_bcg_quality_label.py +++ b/func/Module_bcg_quality_label.py @@ -1030,7 +1030,16 @@ class Data(): artifact_end = append(artifact_end, self.Artifact_a[i]) self.artifact_mask = zeros(len(self.BCG)) for i in range(0, len(self.artifact_number)): - self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 1 + if self.artifact_type[i] == 1: + self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 1 + elif self.artifact_type[i] == 2: + self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 2 + elif self.artifact_type[i] == 3: + self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 3 + elif self.artifact_type[i] == 4: + self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 4 + elif self.artifact_type[i] == 5: + self.artifact_mask[artifact_start[i]: artifact_end[i] + 1] = 5 except Exception as e: return Result().failure(info=Constants.INPUT_FAILURE + Constants.FAILURE_REASON["Get_Artifact_Format_Exception"] + "\n" + format_exc())