Skip to content
This repository has been archived by the owner on Aug 29, 2024. It is now read-only.

Latest commit

 

History

History
92 lines (63 loc) · 3.43 KB

README.zh-CN.md

File metadata and controls

92 lines (63 loc) · 3.43 KB
语言: 🇨🇳 🇺🇸

本仓库不再维护,更新的内容前往: zjykzj/vocdev zjykzj/cocodev

«pnno» 实现了不同格式的数据集或者标注文件的转换

本仓库提供了如下脚本功能:

  1. 将不同数据集转换成算法指定格式。当前已实现
    1. LabelImg标注 -> YoloV5数据格式
    2. VisDrone数据集 -> TLT Kitti数据格式
  2. PyTorch ImageFolder数据集转换为LMDB格式
  3. 在命令行显示图像

内容列表

背景

算法开发过程中经常会涉及到数据集的处理,需要将数据集转换成算法训练时指定的格式,日积月累下来积攒了不少脚本,能不能将这些程序整合在一起,既能够重复使用,也能够降低下一次实现的难度。Let's Do It !!!

安装

$ pip install pnno

使用

数据集/标签文件转换

基本操作:

$ pnno -f <cfg_file>

操作一:转换 tzutalin/labelImg标注文件到 ultralytics/yolov5指定数据集格式。具体配置文件可参考configs/labelimg_2_yolov5.yaml

操作二:转换 VisDrone/VisDrone-Dataset数据集到KITTI标注格式。具体配置文件可参考configs/visdrone_2_tlt.yaml

操作三:转换PyTorch ImageFolder数据集为LMDB格式。参考demo/imagenet_lmdb

更多的使用参考demo/

命令行图像显示

$ simg -f <IMG_FILE>

主要维护人员

  • zhujian - Initial work - zjykzj

参与贡献方式

欢迎任何人的参与!打开issue或提交合并请求。

注意:

许可证

Apache License 2.0 © 2020 zjykzj