typehints, opencv-python-headless

Author: jacobgil

pytorch_grad_cam/base_cam.py

@@ -1,4 +1,4 @@
-from typing import Callable, List, Optional, Tuple
+from typing import Callable, List, Optional, Tuple, Union
 
 import numpy as np
 import torch
@@ -135,7 +135,7 @@ def forward(
 
     def get_target_width_height(
         self, input_tensor: torch.Tensor
-    ) -> Tuple[int, int] | Tuple[int, int, int]:
+    ) -> Union[Tuple[int, int], Tuple[int, int, int]]:
         if len(input_tensor.shape) == 4:
             width, height = input_tensor.size(-1), input_tensor.size(-2)
             return width, height

requirements.txt

@@ -4,6 +4,6 @@ torch>=1.7.1
 torchvision>=0.8.2
 ttach
 tqdm
-opencv-python
+opencv-python-headless
 matplotlib
 scikit-learn