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:
40
scripts/crop_stamp.py
Normal file
40
scripts/crop_stamp.py
Normal file
@@ -0,0 +1,40 @@
|
||||
#!/opt/homebrew/bin/python3.11
|
||||
"""
|
||||
Crop the detected stamp from the image.
|
||||
"""
|
||||
|
||||
from PIL import Image
|
||||
import os
|
||||
|
||||
UUID = "384b0ff44aaaa1f1"
|
||||
BASE_DIR = f"output/{UUID}/florence2_results"
|
||||
IMG_NAME = "raw_6846.jpg"
|
||||
img_path = os.path.join(BASE_DIR, IMG_NAME)
|
||||
|
||||
# Coordinates from the successful run that detected 'stamp'
|
||||
# Format: [x_min, y_min, x_max, y_max]
|
||||
box = [1721.28, 23.22, 1813.44, 173.34]
|
||||
|
||||
print(f"📷 Loading image: {img_path}")
|
||||
if not os.path.exists(img_path):
|
||||
print("❌ Image not found.")
|
||||
exit()
|
||||
|
||||
try:
|
||||
img = Image.open(img_path)
|
||||
print(f"📐 Image Size: {img.width}x{img.height}")
|
||||
|
||||
# Convert float coordinates to int
|
||||
box_int = [int(x) for x in box]
|
||||
print(f"✂️ Cropping box: {box_int}")
|
||||
|
||||
# Crop the image
|
||||
cropped = img.crop(box_int)
|
||||
|
||||
# Save
|
||||
out_path = os.path.join(BASE_DIR, "stamp_crop_detected.jpg")
|
||||
cropped.save(out_path)
|
||||
print(f"✅ Successfully saved cropped stamp to {out_path}")
|
||||
|
||||
except Exception as e:
|
||||
print(f"❌ Error: {e}")
|
||||
Reference in New Issue
Block a user