objectvisions 0.0.1__py3-none-any.whl

objectvisions/__init__.py

@@ -0,0 +1,3 @@
+from .detector import ObjectDetector
+
+__all__ = ["ObjectDetector"]

objectvisions/detector.py

@@ -0,0 +1,42 @@
+import cv2
+import numpy as np
+from .model import load_model
+from .utils import load_class_names
+
+class ObjectDetector:
+    def __init__(self, threshold=0.65, nms_threshold=0.35):
+        self.threshold = threshold
+        self.nms_threshold = nms_threshold
+        
+        self.net = load_model()
+        self.classNames = load_class_names()
+
+    def detect(self, img):
+        classIds, confs, bbox = self.net.detect(img, confThreshold=self.threshold)
+
+        if len(bbox) == 0:
+            return img
+
+        bbox = list(bbox)
+        confs = list(np.array(confs).reshape(1, -1)[0])
+        confs = list(map(float, confs))
+
+        indices = cv2.dnn.NMSBoxes(bbox, confs, self.threshold, self.nms_threshold)
+
+        for i in indices:
+            if len(indices.shape) > 1:
+                i = i[0]
+
+            box = bbox[i]
+            x, y, w, h = box
+            cv2.rectangle(img, (x, y), (x + w, y + h), (0, 255, 0), 2)
+
+            classId = classIds[i][0] if len(classIds[i].shape) > 0 else classIds[i]
+            cv2.putText(img,
+                        self.classNames[classId - 1].upper(),
+                        (x + 10, y + 30),
+                        cv2.FONT_HERSHEY_COMPLEX,
+                        1,
+                        (0, 255, 0),
+                        2)
+        return img

objectvisions/model.py

@@ -0,0 +1,22 @@
+import cv2
+import os
+
+def load_model():
+    base_path = os.path.abspath(os.path.dirname(__file__))
+
+    weightsPath = os.path.join(base_path, "frozen_inference_graph.pb")
+    configPath = os.path.join(base_path, "ssd_mobilenet_v3_large_coco_2020_01_14.pbtxt")
+
+    if not os.path.exists(weightsPath):
+        raise FileNotFoundError(f"Model file not found at {weightsPath}")
+
+    if not os.path.exists(configPath):
+        raise FileNotFoundError(f"Config file not found at {configPath}")
+
+    net = cv2.dnn_DetectionModel(weightsPath, configPath)
+    net.setInputSize(320, 320)
+    net.setInputScale(1.0 / 127.5)
+    net.setInputMean((127.5, 127.5, 127.5))
+    net.setInputSwapRB(True)
+
+    return net

objectvisions/utils.py

@@ -0,0 +1,10 @@
+import os
+
+def load_class_names():
+    base_path = os.path.dirname(__file__)
+    classFile = os.path.join(base_path, "coco.names")
+
+    with open(classFile, 'rt') as f:
+        classNames = f.read().rstrip('\n').split('\n')
+
+    return classNames

objectvisions-0.0.1.dist-info/METADATA

@@ -0,0 +1,51 @@
+Metadata-Version: 2.4
+Name: objectvisions
+Version: 0.0.1
+Summary: Real-time Object Detection using OpenCV DNN
+Author-email: Naman Lohiya <your_email@gmail.com>
+License-Expression: MIT
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: opencv-python
+Requires-Dist: numpy
+Dynamic: license-file
+
+# ObjectVision-Naman 🚀
+
+Real-time Object Detection Library built using OpenCV DNN and SSD MobileNet v3.
+
+## 📌 Features
+
+- Real-time webcam detection
+- Pre-trained COCO model
+- Non-Maximum Suppression (NMS)
+- Easy-to-use ObjectDetector class
+
+## 📦 Installation
+
+```bash
+pip install objectvision-naman
+
+🚀 Usage Example
+import cv2
+from objectvision_naman import ObjectDetector
+
+detector = ObjectDetector()
+
+cap = cv2.VideoCapture(0)
+
+while True:
+    success, img = cap.read()
+    img = detector.detect(img)
+    cv2.imshow("Detection", img)
+
+    if cv2.waitKey(1) & 0xFF == ord('q'):
+        break
+
+🧠 Model Used
+💠SSD MobileNet v3
+💠COCO Dataset
+
+👨‍💻 Author
+Naman Lohiya

objectvisions-0.0.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+objectvisions/__init__.py,sha256=L9URvYsP1NtgoSliFFSDibtPa__9FVD8T0QB1F-BYOQ,68
+objectvisions/detector.py,sha256=SICngLgWibonBAB9fjpOBXm6G7fojPSRrPn_fmR1SOI,1349
+objectvisions/model.py,sha256=FuwydOc_b_0GBXD6ISAbzr5N46WCczwBRmH0bEJYu-4,723
+objectvisions/utils.py,sha256=dLAZ8rg9qFzWpG_V3DyYRTfk-5xpkX3kyNMmd5k4aQU,255
+objectvisions-0.0.1.dist-info/licenses/LICENSE,sha256=2UQ7mgEcHNZ7Dk8pj8BkMO8x-3bW9GYYci4zxiXSVi8,1088
+objectvisions-0.0.1.dist-info/METADATA,sha256=ifvEBmbF25AUCw1Nk4bGDM1RG3tHX3y2To1WKRpooqk,1087
+objectvisions-0.0.1.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
+objectvisions-0.0.1.dist-info/top_level.txt,sha256=0WNsJuMrcp5UrYqr9pKKty9aBYN17WYI72YmxCNW1bg,14
+objectvisions-0.0.1.dist-info/RECORD,,

objectvisions-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

objectvisions-0.0.1.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Naman Lohiya
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

objectvisions-0.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+objectvisions