源码地址:https://github.com/IDEA-Research/DINO
我是在Ubuntu中的Pycharm上面跑的,
首先配置一个新的环境,环境名字就命名为DINO(方法很简单,这里我就不阐述了,不会的小伙伴去看看其他的博文),然后安装好对应的torch和cuda版本(注:cuda最低版本要求为11.0,其他的版本参考源码中的来)

配好新的环境之后,将模型导入至新的环境中,然后安装其他需要的包。
一键安装:pip install -r requirements.txt(如果在安装过程中,有些包安装不上去,可以试试降低包的版本,我当时就是头铁,没注意包的版本问题)
包安装好之后,开始编译环境。在Pycharm中打开终端,然后进入到 models/dino/ops 这个文件目录下: cd models/dino/ops
然后输入: python setup.py build install
输入回车之后可能会出现报错情况:error: could not create 'build': Permission denied,
解决办法:sudo chmod -R 777 xxxxxx(xxxxxx指的是当前的build所在的文件夹的名称,然后输入密码,赋予权限,再次运行就不会报错了)
接着输入 python test.py
环境编译好了,接下来就是准备数据集了,用的是COCO2017数据集,直接去COCO官网上下载(地址:
在Ubuntu和PyCharm中配置DINO模型的步骤与常见问题

文章详细介绍了如何在Ubuntu系统和PyCharm中配置DINO模型的环境,包括创建新环境、安装torch和CUDA(最低要求11.0)、导入模型并安装依赖。在编译环境时,提到了权限问题的解决方案。之后,文章提到使用COCO2017数据集,并提供了数据集的存放方式。在训练过程中,作者遇到并解决了两个问题:模块未找到和CUDA不可用,分别给出了检查CUDA版本和安装nvidia-cuda-toolkit的解决方法。
最低0.47元/天 解锁文章
6470





