Convert color channels.
(rgbs)
| 61 | |
| 62 | |
| 63 | def rgb2bgr(rgbs) -> Union[torch.Tensor, np.ndarray]: |
| 64 | """Convert color channels.""" |
| 65 | bgrs = [rgbs[..., 2, None], rgbs[..., 1, None], rgbs[..., 0, None]] |
| 66 | if isinstance(rgbs, torch.Tensor): |
| 67 | bgrs = torch.cat(bgrs, -1) |
| 68 | elif isinstance(rgbs, np.ndarray): |
| 69 | bgrs = np.concatenate(bgrs, -1) |
| 70 | return bgrs |
| 71 | |
| 72 | |
| 73 | def align_input_to_padded(tensor=Union[List[torch.Tensor], torch.Tensor], |
no test coverage detected