Defrcn代码复现

代码地址
基本环境配置参考这篇文章其中,安装requirement.txt的配置的时候有个报错,把txt文件中的;sklearn换成scikit-learn在运行reqiement就行了。
pytorch和detectron我按照下面的版本不会报错(本人环境py:3.7 cuda:11.0):

conda install pytorch1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch
python -m pip install detectron2 0.3 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu110/torch1.7/index.html

配置完环境下载作者的预训练权重,加入到项目中,加入后修改脚本的几个路径为这个权重的路径。
在这里插入图片描述
然后修改sh脚本中gpu数量,和自己设备不符会报错。】
其余报错可参考这篇文章

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值