Spaces:
Sleeping
Sleeping
File size: 4,049 Bytes
5668d9d 242ac91 5668d9d 242ac91 5668d9d 242ac91 5668d9d 242ac91 5668d9d 242ac91 4eae44a 276ebff 242ac91 276ebff 242ac91 0271310 def047e 242ac91 5668d9d 242ac91 5668d9d 242ac91 5668d9d 242ac91 5668d9d 242ac91 5668d9d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
import streamlit as st
from fengxai.render import setModel, getModel, updateModel, applyBindingController
from fengxai.inference import imagePrediction
from PIL import Image
## model
setModel("distance", 30.07) # (英尺/米)到物体的距离
setModel("sensorHeight", 24) # (毫米)传感器高度
setModel("imageObjectHeight", 2724) # (像素)图像中物体的高度
setModel("lensFocalLength", 60) # (像素)镜头焦距
setModel("imageHeight", 4912) # (像素)图片的高度
## and controller
updateModel( # 物体高度
"objectHeight",
str(getModel("sensorHeight") * getModel("imageObjectHeight") * getModel("distance") / (getModel("lensFocalLength") * getModel("imageHeight")))
)
## (default)set image info
setModel("imageInfo")
def changeUploadFile():
updateModel("imageInfo", {
"imageFile": getModel("imageUploadFile"),
"isUploadImageFile": True # 图片上传开关，控制ai模型在正确的事件下识别图像物体
})
def changeDelete():
updateModel("imageInfo", "")
## view: side 左侧边栏
def render():
st.set_page_config(layout="wide")
with st.sidebar:
st.title("example")
st.image("assets/e2.jpg")
if st.button('use'):
updateModel("imageInfo", {
"imageFile": "assets/e2.jpg",
"isUploadImageFile": True
})
# page
st.title("Rjxai v4_1: :blue[Get object height]")
with st.container():
st.title("Step 1: Rjxai Image Identification")
st.file_uploader(
label = "Upload your image here: png, jpg, jpeg",
type=['png','jpg','jpeg'],
on_change=changeUploadFile,
key="imageUploadFile",
)
if getModel('imageInfo') != "":
def callback(output):
print("output=", output)
updateModel("imageHeight", output["imageHeight"])
updateModel("imageObjectHeight", output["objectHeight"])
c1, c2 = st.columns([0.7, 0.3])
with c1.container():
imageInput = Image.open(output["imageOutput"])
st.image(imageInput)
with c2.container():
st.button("delete image", on_click=changeDelete)
if getModel("imageInfo")["isUploadImageFile"]:
def tmpCallback(output):
updateModel("imageInfo", {
"imageFile": getModel("imageInfo")["imageFile"],
"tmpUploadImageInfo": output, # 存放图片的临时信息
"isUploadImageFile": False
})
print("output=", output)
callback(output)
imagePrediction(getModel('imageInfo')["imageFile"], tmpCallback)
else:
callback(getModel("imageInfo")["tmpUploadImageInfo"])
with st.container():
st.title("Step 2: Calculate real object height")
col1, col2 = st.columns(2)
with col1.container():
st.subheader("Input:")
st.number_input('(pixels) image height', key="imageHeight")
st.number_input('(pixels) image object height', key="imageObjectHeight")
st.number_input('(mm) sensor height', key="sensorHeight")
st.number_input('(mm) lens focal length', key="lensFocalLength")
st.number_input('(feet or meters) distance', key="distance")
with col2:
st.subheader("Output:")
st.text_area("(feet or meters) object height:", key="objectHeight", disabled=True)
# control 绑定
applyBindingController(render)
|