Bug fix

HHousen · Jul 25, 2023 · 7801269 · 7801269
1 parent 7e04d0e
commit 7801269
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lecture2notes/end_to_end/figure_detection.py b/lecture2notes/end_to_end/figure_detection.py
@@ -341,7 +341,10 @@ def detect_figures(
  # If `entropy_check` is a boolean, then set it to the default
  if type(entropy_check) is bool and entropy_check:
  entropy_check = 2.5
- gray = cv2.cvtColor(figure, cv2.COLOR_BGR2GRAY)
+ try:
+ gray = cv2.cvtColor(figure, cv2.COLOR_BGR2GRAY)
+ except:
+ continue
  high_entropy = shannon_entropy(gray) > entropy_check
  if not high_entropy:
  continue