python 通过使用Yolact训练数据集


Posted in Python onApril 06, 2021

可能是由于yolact官方更新过其项目代码,所以网上其他人的yolact训练使用的config文件和我的稍微有区别。但总体还是差不多的。

1:提前准备好自己的数据集

使用labelme来制作分割数据集,但是得到的是一个个单独的json文件。需要将其转换成coco。
labelme2coco.py如下所示(代码来源:github):

import os
import json
import numpy as np
import glob
import shutil
from sklearn.model_selection import train_test_split
np.random.seed(41)

#0为背景,此处根据你数据集的类别来修改key
classname_to_id = {"1": 1}

class Lableme2CoCo:

 def __init__(self):
  self.images = []
  self.annotations = []
  self.categories = []
  self.img_id = 0
  self.ann_id = 0

 def save_coco_json(self, instance, save_path):
  json.dump(instance, open(save_path, 'w', encoding='utf-8'), ensure_ascii=False, indent=1) # indent=2 更加美观显示

 # 由json文件构建COCO
 def to_coco(self, json_path_list):
  self._init_categories()
  for json_path in json_path_list:
   obj = self.read_jsonfile(json_path)
   self.images.append(self._image(obj, json_path))
   shapes = obj['shapes']
   for shape in shapes:
    annotation = self._annotation(shape)
    self.annotations.append(annotation)
    self.ann_id += 1
   self.img_id += 1
  instance = {}
  instance['info'] = 'spytensor created'
  instance['license'] = ['license']
  instance['images'] = self.images
  instance['annotations'] = self.annotations
  instance['categories'] = self.categories
  return instance

 # 构建类别
 def _init_categories(self):
  for k, v in classname_to_id.items():
   category = {}
   category['id'] = v
   category['name'] = k
   self.categories.append(category)

 # 构建COCO的image字段
 def _image(self, obj, path):
  image = {}
  from labelme import utils
  img_x = utils.img_b64_to_arr(obj['imageData'])
  h, w = img_x.shape[:-1]
  image['height'] = h
  image['width'] = w
  image['id'] = self.img_id
  image['file_name'] = os.path.basename(path).replace(".json", ".jpg")
  return image

 # 构建COCO的annotation字段
 def _annotation(self, shape):
  label = shape['label']
  points = shape['points']
  annotation = {}
  annotation['id'] = self.ann_id
  annotation['image_id'] = self.img_id
  annotation['category_id'] = int(classname_to_id[label])
  annotation['segmentation'] = [np.asarray(points).flatten().tolist()]
  annotation['bbox'] = self._get_box(points)
  annotation['iscrowd'] = 0
  annotation['area'] = 1.0
  return annotation

 # 读取json文件,返回一个json对象
 def read_jsonfile(self, path):
  with open(path, "r", encoding='utf-8') as f:
   return json.load(f)

 # COCO的格式: [x1,y1,w,h] 对应COCO的bbox格式
 def _get_box(self, points):
  min_x = min_y = np.inf
  max_x = max_y = 0
  for x, y in points:
   min_x = min(min_x, x)
   min_y = min(min_y, y)
   max_x = max(max_x, x)
   max_y = max(max_y, y)
  return [min_x, min_y, max_x - min_x, max_y - min_y]


if __name__ == '__main__':
 labelme_path = "labelme/" # 此处根据你的数据集地址来修改
 saved_coco_path = "./"
 # 创建文件
 if not os.path.exists("%scoco/annotations/"%saved_coco_path):
  os.makedirs("%scoco/annotations/"%saved_coco_path)
 if not os.path.exists("%scoco/images/train2017/"%saved_coco_path):
  os.makedirs("%scoco/images/train2017"%saved_coco_path)
 if not os.path.exists("%scoco/images/val2017/"%saved_coco_path):
  os.makedirs("%scoco/images/val2017"%saved_coco_path)
 # 获取images目录下所有的joson文件列表
 json_list_path = glob.glob(labelme_path + "/*.json")
 # 数据划分,这里没有区分val2017和tran2017目录,所有图片都放在images目录下
 train_path, val_path = train_test_split(json_list_path, test_size=0.12)
 print("train_n:", len(train_path), 'val_n:', len(val_path))

 # 把训练集转化为COCO的json格式
 l2c_train = Lableme2CoCo()
 train_instance = l2c_train.to_coco(train_path)
 l2c_train.save_coco_json(train_instance, '%scoco/annotations/instances_train2017.json'%saved_coco_path)
 for file in train_path:
  shutil.copy(file.replace("json","jpg"),"%scoco/images/train2017/"%saved_coco_path)
 for file in val_path:
  shutil.copy(file.replace("json","jpg"),"%scoco/images/val2017/"%saved_coco_path)

 # 把验证集转化为COCO的json格式
 l2c_val = Lableme2CoCo()
 val_instance = l2c_val.to_coco(val_path)
 l2c_val.save_coco_json(val_instance, '%scoco/annotations/instances_val2017.json'%saved_coco_path)

只需要修改两个地方即可,然后放到data文件夹下。
最后,得到的coco格式的数据集如下所示:

python 通过使用Yolact训练数据集

至此,数据准备已经结束。

2:下载github存储库

网址:YOLACT

之后解压,但是我解压的时候不知道为啥没有yolact.py这个文件。后来又建了一个py文件,复制了里面的代码。

python 通过使用Yolact训练数据集

下载权重文件,把权重文件放到yolact-master下的weights文件夹里(没有就新建):

python 通过使用Yolact训练数据集

python 通过使用Yolact训练数据集

3:修改config.py

文件所在位置:

python 通过使用Yolact训练数据集

修改类别,把原本的coco的类别全部注释掉,修改成自己的(如红色框),注意COCO_CLASSES里有一个逗号。

python 通过使用Yolact训练数据集

修改数据集地址dataset_base

python 通过使用Yolact训练数据集

修改coco_base_config(下面第二个横线max_iter并不是控制训练轮数的,第二张图中的max_iter才是)

python 通过使用Yolact训练数据集

python 通过使用Yolact训练数据集

4:训练

cd到指定路径下,执行下面命令即可

python train.py --config=yolact_base_config

刚开始:

python 通过使用Yolact训练数据集

因为我是租的云服务器,在jupyter notebook里训练的。输出的训练信息比较乱。

训练几分钟后:

python 通过使用Yolact训练数据集

主要看T后面的数字即可,好像他就是总的loss,如果它收敛了,按下Ctrl+C,即可中止训练,保存模型权重。

第一个问题:

PytorchStreamReader failed reading zip archive: failed finding central directory

python 通过使用Yolact训练数据集

第二个问题:
(但是不知道为啥,我训练时如果中断,保存的模型不能用来测试,会爆出下面的错误)

RuntimeError: unexpected EOF, expected *** more bytes. The file might be corruptrd

没办法解决,所以只能跑完,自动结束之后保存的模型拿来测试(自动保存的必中断保存的要大十几兆)

模型保存的格式:<config>_<epoch>_<iter>.pth。如果是中断的:<config>_<epoch>_<iter>_interrupt.pth

5:测试

使用官网的测试命令即可

python 通过使用Yolact训练数据集

以上就是python 使用Yolact训练自己的数据集的详细内容,更多关于python 训练数据集的资料请关注三水点靠木其它相关文章!

Python 相关文章推荐
python Django连接MySQL数据库做增删改查
Nov 07 Python
详解Python网络爬虫功能的基本写法
Jan 28 Python
Flask框架的学习指南之制作简单blog系统
Nov 20 Python
Python爬虫实例_城市公交网络站点数据的爬取方法
Jan 10 Python
python使用turtle绘制分形树
Jun 22 Python
Flask模拟实现CSRF攻击的方法
Jul 24 Python
Python3中关于cookie的创建与保存
Oct 21 Python
pytorch实现mnist数据集的图像可视化及保存
Jan 14 Python
python3利用Axes3D库画3D模型图
Mar 25 Python
python中format函数如何使用
Jun 22 Python
selenium+python实现基本自动化测试的示例代码
Jan 27 Python
Python爬虫基础之初次使用scrapy爬虫实例
Jun 26 Python
python生成随机数、随机字符、随机字符串
Apr 06 #Python
Django项目配置Memcached和Redis, 缓存选择哪个更有优势
Apr 06 #Python
PySwarms(Python粒子群优化工具包)的使用:GlobalBestPSO例子解析
python实现批量提取指定文件夹下同类型文件
Apr 05 #Python
python实现ROA算子边缘检测算法
python实现批量移动文件
Python list去重且保持原顺序不变的方法
Apr 03 #Python
You might like
解决php中Cannot send session cache limiter 的问题的方法
2007/04/27 PHP
php foreach 使用&amp;(与运算符)引用赋值要注意的问题
2010/02/16 PHP
使用PHPCMS搭建wap手机网站
2015/09/20 PHP
Laravel 使用查询构造器配合原生sql语句查询的例子
2019/10/12 PHP
给Function做的OOP扩展
2009/05/07 Javascript
jquery对象和DOM对象的区别介绍
2013/08/09 Javascript
tangram框架响应式加载图片方法
2013/11/21 Javascript
jsPDF导出pdf示例
2014/05/02 Javascript
教你如何自定义百度分享插件以及bshare分享插件的分享按钮
2014/06/20 Javascript
JavaScript插件化开发教程 (三)
2015/01/27 Javascript
JavaScript实现的encode64加密算法实例分析
2015/04/15 Javascript
每日十条JavaScript经验技巧(二)
2016/06/23 Javascript
javascript函数中的3个高级技巧
2016/09/22 Javascript
基于JavaScript实现移动端无限加载分页
2017/03/27 Javascript
Require.JS中的几种define定义方式示例
2017/06/01 Javascript
Angular指令之restict匹配模式的详解
2017/07/27 Javascript
js实现音乐播放控制条
2017/09/09 Javascript
详解在React里使用&quot;Vuex&quot;
2018/04/02 Javascript
多个Vue项目部署到服务器的步骤记录
2020/10/22 Javascript
解决iview table组件里的 固定列 表格不自适应的问题
2020/11/13 Javascript
[01:14:12]2018DOTA2亚洲邀请赛4.7 总决赛 LGD vs Mineski 第二场
2018/04/09 DOTA
python求质数列表的例子
2019/11/24 Python
Django后端发送小程序微信模板消息示例(服务通知)
2019/12/17 Python
Tensorflow实现部分参数梯度更新操作
2020/01/23 Python
Python解释器及PyCharm工具安装过程
2020/02/26 Python
利用OpenCV中对图像数据进行64F和8U转换的方式
2020/06/03 Python
CSS Grid布局教程之什么是网格布局
2014/12/30 HTML / CSS
科颜氏加拿大官方网站: Kiehl’s加拿大
2016/08/16 全球购物
英国领先的狗和宠物美容专家:Christies Direct
2017/04/03 全球购物
Champs Sports加拿大:北美最大的以商场为基础的专业运动鞋和服装零售商之一
2018/05/01 全球购物
IT工程师岗位职责
2014/07/04 职场文书
2014年行政执法工作总结
2014/12/11 职场文书
创业计划书之餐饮馄饨店
2019/07/18 职场文书
优秀范文:《但愿人长久》教学反思3篇
2019/10/24 职场文书
使用Redis实现分布式锁的方法
2022/06/16 Redis
maven 解包依赖项中的文件的解决方法
2022/07/15 Java/Android