一、项目窗口展示
原代码和数据集地址:https://m.tb.cn/h.g8V294l?tk=ssHf3cwymML
二、窗口可视化(window.py)
运行window.py即可出现上述可视化窗口
在代码第83行
self.model = self.model_load(weights="runs/train/exp2/weights/best.pt",
device=self.device) # todo 指明模型加载的位置的设备
weights =“” :这里需要改成你的模型权重路径,这样就可以通过图形化界面,对新的图片进行舌诊预测和分割。
该图形化界面还有一个视频舌诊功能
只要点击下述按钮就可进行检测,甚至可以用电脑摄像头检测。
三、训练集
原代码和数据集地址:https://m.tb.cn/h.g8V294l?tk=ssHf3cwymML