notid/ai-game-2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

progress

Browse Source
This commit is contained in:
Bryce
2026-03-04 22:05:05 -08:00
parent 33dc00cb6a
commit 7c0ff65a19
10 changed files with 777 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tools/mask_to_polygon.py
View File

@@ -257,8 +257,8 @@ def main():
parser.add_argument(
"--min-area",
type=int,
default=100,
help="Minimum contour area to include in multiple mode (default: 100)",
default=150,
help="Minimum contour area to include (default: 150)",
)
args = parser.parse_args()
@@ -279,13 +279,14 @@ def main():
print("Error: No contours found in mask", file=sys.stderr)
sys.exit(1)
contours = [c for c in contours if cv2.contourArea(c) >= args.min_area]
if not contours:
print("Error: No contours meet minimum area requirement", file=sys.stderr)
sys.exit(1)
if args.mode == "multiple":
contours = sorted(contours, key=cv2.contourArea, reverse=True)
contours = [c for c in contours if cv2.contourArea(c) >= args.min_area]
if not contours:
print("Error: No contours meet minimum area requirement", file=sys.stderr)
sys.exit(1)
output_base = args.output if args.output else args.image.with_suffix("")
output_dir = output_base.parent