feat: update Python processors and add utility scripts
- Update ASR, face, OCR, pose processors - Add release pre-flight check script - Add synonym generation, chunk processing scripts - Add face recognition, stamp search utilities
This commit is contained in:
45
scripts/visualize_stamp.py
Normal file
45
scripts/visualize_stamp.py
Normal file
@@ -0,0 +1,45 @@
|
||||
#!/opt/homebrew/bin/python3.11
|
||||
"""
|
||||
Draw Detection Result on Image
|
||||
"""
|
||||
|
||||
import cv2
|
||||
import os
|
||||
|
||||
UUID = "384b0ff44aaaa1f1"
|
||||
OUTPUT_DIR = f"output/{UUID}/florence2_results"
|
||||
INPUT_IMG = os.path.join(OUTPUT_DIR, f"raw_6846.jpg")
|
||||
OUTPUT_IMG = os.path.join(OUTPUT_DIR, f"raw_6846_detected.jpg")
|
||||
|
||||
# Florence-2 Result from previous run: [x_min, y_min, x_max, y_max]
|
||||
# Note: Florence-2 usually returns coordinates in normalized 0-1000 scale.
|
||||
box_raw = [1721.28, 23.22, 1813.44, 173.34]
|
||||
|
||||
# Image dimensions
|
||||
img_width, img_height = 1920, 1080
|
||||
|
||||
print(f"🎨 Drawing box on {INPUT_IMG}...")
|
||||
|
||||
img = cv2.imread(INPUT_IMG)
|
||||
if img is None:
|
||||
print("❌ Failed to load image.")
|
||||
exit()
|
||||
|
||||
# Convert normalized coordinates to pixel coordinates
|
||||
# Florence-2 uses 1000x1000 normalized coordinates
|
||||
x1 = int((box_raw[0] / 1000.0) * img_width)
|
||||
y1 = int((box_raw[1] / 1000.0) * img_height)
|
||||
x2 = int((box_raw[2] / 1000.0) * img_width)
|
||||
y2 = int((box_raw[3] / 1000.0) * img_height)
|
||||
|
||||
print(f"📍 Pixel Coordinates: ({x1}, {y1}) to ({x2}, {y2})")
|
||||
|
||||
# Draw Rectangle (Color: Green, Thickness: 4)
|
||||
cv2.rectangle(img, (x1, y1), (x2, y2), (0, 255, 0), 4)
|
||||
|
||||
# Add Label
|
||||
cv2.putText(img, "STAMP", (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
|
||||
|
||||
# Save
|
||||
cv2.imwrite(OUTPUT_IMG, img)
|
||||
print(f"✅ Saved result to {OUTPUT_IMG}")
|
||||
Reference in New Issue
Block a user