TheKnight115 commited on
Commit
1cf4588
·
verified ·
1 Parent(s): 3889969

Update processor.py

Browse files
Files changed (1) hide show
  1. processor.py +2 -8
processor.py CHANGED
@@ -184,10 +184,7 @@ def process_video(video_path, font_path, violation_image_path='violation.jpg'):
184
  license_plate_pil = Image.fromarray(cv2.cvtColor(license_crop, cv2.COLOR_BGR2RGB))
185
  temp_image_path = 'license_plate.png'
186
  license_plate_pil.save(temp_image_path)
187
- # Placeholder for OCR
188
- # license_plate_text = model_ocr.chat(processor, temp_image_path, ocr_type='ocr')
189
- # For demonstration, we'll mock the OCR result
190
- license_plate_text = "1234AB"
191
  filtered_text = filter_license_plate_text(license_plate_text)
192
 
193
  if filtered_text:
@@ -268,10 +265,7 @@ def process_image(image_path, font_path, violation_image_path='violation.jpg'):
268
  license_plate_pil = Image.fromarray(cv2.cvtColor(license_crop, cv2.COLOR_BGR2RGB))
269
  temp_image_path = 'license_plate.png'
270
  license_plate_pil.save(temp_image_path)
271
- # Placeholder for OCR
272
- # license_plate_text = model_ocr.chat(processor, temp_image_path, ocr_type='ocr')
273
- # For demonstration, we'll mock the OCR result
274
- license_plate_text = "1234AB"
275
  filtered_text = filter_license_plate_text(license_plate_text)
276
 
277
  if filtered_text:
 
184
  license_plate_pil = Image.fromarray(cv2.cvtColor(license_crop, cv2.COLOR_BGR2RGB))
185
  temp_image_path = 'license_plate.png'
186
  license_plate_pil.save(temp_image_path)
187
+ license_plate_text = model_ocr.chat(processor, temp_image_path, ocr_type='ocr')
 
 
 
188
  filtered_text = filter_license_plate_text(license_plate_text)
189
 
190
  if filtered_text:
 
265
  license_plate_pil = Image.fromarray(cv2.cvtColor(license_crop, cv2.COLOR_BGR2RGB))
266
  temp_image_path = 'license_plate.png'
267
  license_plate_pil.save(temp_image_path)
268
+ license_plate_text = model_ocr.chat(processor, temp_image_path, ocr_type='ocr')
 
 
 
269
  filtered_text = filter_license_plate_text(license_plate_text)
270
 
271
  if filtered_text: