本期给大家介绍的是129基于FCN图像分割算法检测火焰_数据集json转mask,效果图如下:
代码仓库和视频演示地址:
129基于FCN图像分割算法检测火焰_数据集json转mask_哔哩哔哩_bilibili
代码整体是非常简便的,主要是dataset数据集文件夹和3个py文件。
运行01json转mask格式.py能将labelme打标签好的json格式文件变成mask图,并划分训练集和验证集,如果数据集本身是mask的可直接训练。
运行02train.py会对训练集和验证集的数据进行读取,训练模型,训练好的模型保存在本地。并生成评价指标图。
运行03pyqt.py可以选择自己感兴趣的图片进行识别,通过调用上一步训练好的模型识别图片中的火焰。