修复了<BCG的质量标注>中体动显示异常的问题

This commit is contained in:
2025-05-29 21:21:02 +08:00
parent e379aa6a4a
commit afc93cf498

View File

@ -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())