Fix exception (#183)

* Fixing exception in `LoadImagesAndLabels` * Fixing exception in thop importing * Fixing exception in `exif_size`
zhiqwang · Oct 2, 2021 · f0746c6 · f0746c6
1 parent c9b5b6f
commit f0746c6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/yolort/v5/utils/datasets.py b/yolort/v5/utils/datasets.py
@@ -74,7 +74,7 @@ def exif_size(img):
             s = (s[1], s[0])
         elif rotation == 8:  # rotation 90
             s = (s[1], s[0])
-    except KeyError:
+    except AttributeError:
         pass
 
     return s
@@ -368,7 +368,7 @@ def __init__(
             assert cache["version"] == 0.4 and cache["hash"] == get_hash(
                 self.label_files + self.img_files
             )
-        except AssertionError:
+        except FileNotFoundError:
             cache, exists = self.cache_labels(cache_path, prefix), False  # cache
 
         # Display cache

diff --git a/yolort/v5/utils/torch_utils.py b/yolort/v5/utils/torch_utils.py
@@ -155,11 +155,11 @@ def profile(input, ops, n=10, device=None):
             )
             # dt forward, backward
             tf, tb, t = 0.0, 0.0, [0.0, 0.0, 0.0]
-            try:
+            if thop is None:
+                flops = 0
+            else:
                 # GFLOPs
                 flops = thop.profile(m, inputs=(x,), verbose=False)[0] / 1e9 * 2
-            except ImportError:
-                flops = 0
 
             try:
                 for _ in range(n):