ObjectDatasetTools是微软官方开源的singleshotpose提供的linemode数据集制作工具,本文是根据自己改良版的6d状态估计环境下搭配,和官方提供有些出处(官方的毕竟是2018年的老古董,不改良很多包用不了)
一,创建ObjectDatasetTools环境
windows安装好anaconda和cuda,cudnn(可以看我之前的文章)后,打开cmd终端创建yolo6d环境
二,安装pytorch
创建完成进入环境内
pytorch官网
# CUDA 12.1
输入conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia
安装与cuda12.1对应版本的pytorch(下载太慢挂梯子或者换源都行)
耐心等待下载
输入python,输入import torch,输入torch.cuda.is_available()来检查gpu是否可以,确保torch安装成功
(注:查看torch版本是否有+cuxxx,若没有,查看我发的另一篇torch安装教程)
三,配置ObjectDatasetTools环境(非官方)
python3.8版本torch=2.2.2 CUDA12.1 CUDNN9.1.0
输入pip install --upgrade pip(我这里把梯子关了,不然会warning报错,没报就继续)更新pip,如果不行就试试python -m pip install --upgrade pip或者python -m pip install --upgrade pip --user给权限
pip3 install opencv-python==4.6.0.66 opencv-contrib-python==4.6.0.66 scipy==1.3.3 numpy==1.23.1 matplotlib==3.2.2 Cython==3.0.10 trimesh==2.38.24 scikit-image==0.17.2 scikit-learn==1.0.2 open3d==0.10.0 plyfile shapely pypng tqdm pykdtree -i https://pypi.tuna.tsinghua.edu.cn/simple
之前用的open3d==0.10.0(后面补一篇安装教程)但是安装时报错了(已解决参考另一篇文章方法二补充),没加版本限制可以正常安装
从github下载ObjectDatasetTools
修改create_lable_files.py 153行(OPENCV版本太高)
_, contours, _ = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,
改为contours, _ = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,
不出意外,环境搭配完成