ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

用PVN3D训练自制LINEMOD数据集过程记录

2022-02-23 19:01:02  阅读:419  来源: 互联网

标签:自制 录制 yml train PVN3D txt 数据 LINEMOD


1.制作自己的LINEMOD数据集

(1)制作过程与步骤

制作数据集的过程参考:6d位姿估计—制作自己的LINEMOD数据集过程记录(有效的方法)

(2)合并数据集、生成test.txt与train.txt

数据制作过程中在获取帧变换(python compute_gt_poses.py LINEMOD/duck2
)步骤中,若录制的图片过多的话,内存会溢出,因此不建议一次性录制太多图片,本人在record2.py中,RECORD_LENGTH=50,每次录制大概能够生成一千零几张图片。

基于以上原因,我们需要录制多个数据集,并在后期需要将rgb、depth、mask图像以及gt.yml、info.yml进行合并,并生成train.txt和test.txt。合并程序的电脑不能联网,有需要的可以私信我。

2.将PVN3D的输入替换为自己的数据集

(1)将datasets/linemod/Linemod_preprocessed/data/09改为009,将自己的merge文件夹改为09;

(2)将JPEGImages改为rgb;

(3)在common.py中修改相机内参;并将第37行的mini_batch_size=64

注:因为在生成info.yml的时候depth_scale用的是0.001

3.训练

cd pvn3d
python3 -m train.train_linemod_pvn3d --cls duck

标签:自制,录制,yml,train,PVN3D,txt,数据,LINEMOD
来源: https://blog.csdn.net/weixin_41837701/article/details/123094114

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有