代码下载和视频演示地址:
058基于python深度学习AI的车辆车高_车宽_横截面积检测_哔哩哔哩_bilibili
效果图如下:
代码所有文件:
运行01create_txt.py会将dataset文件下的图片路径及标签保存在txt文本内,
运行02train.py会对图片进行读取并训练模型保存在weights文件下,
运行03predict.py会对单张图片进行预测,
运行04pyqt界面.py可以展示一个pyqt的可视化交互界面,通过点击按钮加载感兴趣的图片进行识别。