Add object detection demo

wtffqbpl · Feb 8, 2025 · 13e0a7d · 13e0a7d
1 parent d9b353c
commit 13e0a7d
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/computer_vision/catdog.png b/computer_vision/catdog.png
diff --git a/computer_vision/object_detection.py b/computer_vision/object_detection.py
@@ -0,0 +1,26 @@
+#! coding: utf-8
+
+
+import os
+import unittest
+import torch
+import torchvision
+import sys
+from pathlib import Path
+sys.path.append(str(Path(__file__).resolve().parent.parent))
+from utils.plot import ImageUtils
+
+
+class IntegrationTest(unittest.TestCase):
+    def setUp(self) -> None:
+        self.img_path = os.path.join(str(Path(__file__).resolve().parent), 'catdog.png')
+        pass
+
+    def test_simple(self):
+        img = ImageUtils.open(self.img_path)
+        ImageUtils.imshow(img)
+    pass
+
+
+if __name__ == "__main__":
+    unittest.main(verbosity=True)