v3.0.72

rectlabs · Aug 16, 2023 · 33d3622 · 33d3622
1 parent 8a30580
commit 33d3622
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = rectvision
-version = 3.0.71
+version = 3.0.72
 author = Rectlabs
 author_email = rectanglenet@gmail.com
 description = Rectvision package

diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 with codecs.open(os.path.join(here, "README.md"), encoding="utf-8") as fh:
     long_description = "\n" + fh.read()
 
-VERSION = '3.0.71'
+VERSION = '3.0.72'
 DESCRIPTION = 'A low-code tool to help create your own AI'
 LONG_DESCRIPTION = 'A package that allows to build your own computer visions and NLP systems'
 

diff --git a/src/rectvision/models/yolov5.py b/src/rectvision/models/yolov5.py
@@ -129,14 +129,14 @@ def get_map(self):
           print(process.stderr)
 
 
-    def inference(self, images, out_dir, img_size = 416, confidence= 0.3, iou_thres = 0.3):
+    def inference(self, image, out_dir, img_size = 416, confidence= 0.3, iou_thres = 0.3):
       self.outdir =out_dir + '/' + otp()
       self.detect_model = os.path.join(self.project_dir, "yolov5/detect.py")
       self.model_weights = os.path.join(self.project_dir, "yolov5/runs/train", self.project_name, "weights/best.pt")
       print('Inference in progress...')
       process = subprocess.run(["python", self.detect_model, 
-                        "--source", images,
-                        "--img",  img_size,
+                        "--source", image,
+                        "--img",  str(img_size),
                         "--weights", self.model_weights,
                         "--conf", str(confidence),
                         "--iou-thres", str(iou_thres),