(seq, patterns)
| 80 | |
| 81 | |
| 82 | def rgb2bayer_seq(seq, patterns): |
| 83 | f, h, w, c = seq.shape |
| 84 | assert f == len(patterns) |
| 85 | assert c == 3 |
| 86 | h = h // 2 * 2 |
| 87 | w = w // 2 * 2 |
| 88 | bayer_masks = {pattern: rgb_bayer_masks((h, w), pattern) for pattern in bayer_patterns} |
| 89 | seq_masks = [bayer_masks[pattern] for pattern in patterns] |
| 90 | reds, greens, blues = [np.stack(channel) for channel in zip(*seq_masks)] |
| 91 | return seq[:, :h, :w, 0] * reds + seq[:, :h, :w, 1] * greens + seq[:, :h, :w, 2] * blues |
| 92 | |
| 93 | |
| 94 | def conv2d_border101(img, filt): |
no test coverage detected