feat: ETERNAL rating class support

This commit is contained in:
2024-03-20 15:53:10 +08:00
parent 17f6c2bac7
commit 00cd32dfdc
3 changed files with 19 additions and 0 deletions

View File

@ -80,6 +80,7 @@ class DeviceOcr:
self.masker.rating_class_prs(roi),
self.masker.rating_class_ftr(roi),
self.masker.rating_class_byd(roi),
self.masker.rating_class_etr(roi),
]
return max(enumerate(results), key=lambda i: np.count_nonzero(i[1]))[0]