windows配置ObjectDatasetTools环境(python3.8)

ObjectDatasetTools是微软官方开源的singleshotpose提供的linemode数据集制作工具,本文是根据自己改良版的6d状态估计环境下搭配,和官方提供有些出处(官方的毕竟是2018年的老古董,不改良很多包用不了)

一,创建ObjectDatasetTools环境

windows安装好anaconda和cuda,cudnn(可以看我之前的文章)后,打开cmd终端创建yolo6d环境

二,安装pytorch

创建完成进入环境内

pytorch官网

Previous PyTorch Versions | PyTorch Installing previous versions of PyTorchhttps://pytorch.org/get-started/previous-versions/

# 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

https://github.com/F2Wang/ObjectDatasetTools/tree/masterhttps://github.com/F2Wang/ObjectDatasetTools/tree/master

修改create_lable_files.py 153行(OPENCV版本太高)
 _, contours, _ = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,    
 改为contours, _ = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,

不出意外,环境搭配完成

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值