Commit ae6d761

committed

add refactor for faster work code

1 parent 30746ff commit ae6d761Copy full SHA for ae6d761

File tree

+1663

-233

lines changed

+1663

-233

lines changed

Lines changed: 8 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -85,14 +85,20 @@ def detect_objects(self, frame):`
`85`	`85`	`bboxes = []`
`86`	`86`	`for result in results:`
`87`	`87`	`for box in result.boxes:`
	`88`	`+ # Get the class ID and bounding box coordinates`
	`89`	`+ cls_id = int(box.cls[0]) # Class ID`
`88`	`90`	`x1, y1, x2, y2 = box.xyxy[0]`
`89`	`91`	`w = int(x2 - x1)`
`90`	`92`	`h = int(y2 - y1)`
`91`		`- if w > 0 and h > 0:`
	`93`	`+`
	`94`	`+ # Filter for class 0 only`
	`95`	`+ if cls_id == 0 and w > 0 and h > 0:`
`92`	`96`	`bboxes.append((int(x1), int(y1), w, h))`
	`97`	`+ elif cls_id != 0:`
	`98`	`+ logging.debug(f"Skipping detection for class {cls_id}.")`
`93`	`99`	`else:`
`94`	`100`	`logging.warning(f"Detected invalid bounding box with zero area: {(x1, y1, w, h)}")`
`95`		`- logging.debug(f"Detected {len(bboxes)} bounding boxes.")`
	`101`	`+ logging.debug(f"Detected {len(bboxes)} bounding boxes for class 0.")`
`96`	`102`	`return bboxes`
`97`	`103`	`except Exception as e:`
`98`	`104`	`logging.error(f"Error during object detection: {e}")`

67.6 KB

Binary file not shown.

Comments

(0)