add multiple anchors support: use np.all

This commit is contained in:
Levi Vasconcelos committed 2024-02-26 17:41:10 -03:00
1 parent d63837181f
commit e8ab777dfd
1 file changed
+1 -1
+1 -1
View File
@@ -69,7 +69,7 @@ class PolygonZone:
)
is_in_zone = self.mask[all_clipped_anchors[:, :, 1], all_clipped_anchors[:, :, 0]].transpose().astype(bool)
is_in_zone = np.sum(is_in_zone, axis=1) == is_in_zone.shape[1]
is_in_zone = np.all(is_in_zone, axis=1)
self.current_count = int(np.sum(is_in_zone))
return is_in_zone.astype(bool)