129基于FCN图像分割算法检测火焰_数据集json转mask

本文介绍了如何使用129基于FCN的图像分割算法进行火焰检测,包括将labelme的json格式数据转换为mask,划分训练集和验证集,以及使用PyTorch训练和测试模型的过程,同时提供了代码下载和视频演示链接。
摘要由CSDN通过智能技术生成

 本期给大家介绍的是129基于FCN图像分割算法检测火焰_数据集json转mask,效果图如下:

代码下载和视频演示地址:

129基于FCN图像分割算法检测火焰_数据集json转mask_哔哩哔哩_bilibili

代码整体是非常简便的,主要是dataset数据集文件夹和3个py文件。

运行01json转mask格式.py能将labelme打标签好的json格式文件变成mask图,并划分训练集和验证集,如果数据集本身是mask的可直接训练。

运行02train.py会对训练集和验证集的数据进行读取,训练模型,训练好的模型保存在本地。并生成评价指标图。

运行03pyqt.py可以选择自己感兴趣的图片进行识别,通过调用上一步训练好的模型识别图片中的火焰。

注:

下载本代码环境自行安装

安装python和pytorch即可运行,

安装环境可

  • 11
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值