Vahe commited on
Commit
1739d0b
1 Parent(s): a473c29

input details showing part added

Browse files
Files changed (1) hide show
  1. extractor.py +3 -0
extractor.py CHANGED
@@ -2,6 +2,7 @@ import tensorflow as tf
2
  import numpy as np
3
  from PIL import Image
4
  import cv2
 
5
 
6
  def get_card_xy(model_path, image_path):
7
  #model_path = 'odo_detector.tflite'
@@ -11,6 +12,8 @@ def get_card_xy(model_path, image_path):
11
  input_details = interpreter.get_input_details()
12
  output_details = interpreter.get_output_details()
13
 
 
 
14
  # Obtain the height and width of the corresponding image from the input tensor
15
  image_height = input_details[0]['shape'][1] # 640
16
  image_width = input_details[0]['shape'][2] # 640
 
2
  import numpy as np
3
  from PIL import Image
4
  import cv2
5
+ import streamlit as st
6
 
7
  def get_card_xy(model_path, image_path):
8
  #model_path = 'odo_detector.tflite'
 
12
  input_details = interpreter.get_input_details()
13
  output_details = interpreter.get_output_details()
14
 
15
+ st.write(f"{input_details}")
16
+
17
  # Obtain the height and width of the corresponding image from the input tensor
18
  image_height = input_details[0]['shape'][1] # 640
19
  image_width = input_details[0]['shape'][2] # 640