昨天深夜调试模型导出时,突然接到产品经理电话:“明早要给客户演示RT-DETR的实时检测效果,能不能做个网页版?”
我看了眼时间——凌晨1点。
传统Web开发流程显然来不及,这时候就该祭出Streamlit这个神器了。
从命令行到网页的魔法
我们训练好的RT-DETR模型通常活在Python脚本里,用OpenCV读取摄像头,终端打印检测结果。
但客户想看的是能交互的网页界面。传统方案要搞Flask后端、HTML前端、WebSocket通信……一夜根本不可能完成。
Streamlit的妙处在于,它让你用写Python脚本的方式直接生成Web应用。
下面这个例子,我用了不到30分钟就搭出了原型:
importstreamlitasstimportcv2importtorchfromPILimportImageimportnumpyasnp