- for k in range(2, 2 * marker_size - 3):
- i = k + 1 - marker_size
- j = marker_size - 2 - abs(k - marker_size + 1)
- separator[:, :, separator.size(2) // 2 + i, j] = 0
- separator[:, :, separator.size(2) // 2 + i + 1, j] = 0
+ separator[:, :, 0] = 0
+ separator[:, :, h - 1] = 0
+
+ for k in range(1, 2 * marker_size - 8):
+ i = k - (marker_size - 4)
+ j = marker_size - 5 - abs(i)
+ separator[:, :, h // 2 - 1 + i, 2 + j] = 0
+ separator[:, :, h // 2 - 1 + i + 1, 2 + j] = 0