Detectron2:如何启动项目?

问题描述 投票:0回答:1

现在我有一个已经标记为COCO格式的数据集,从github下载Detectron2的代码后该怎么办?

我现在有了torch和torchvision,似乎我需要编写一个.py文件,那么我应该在这个文件中包含什么?文件太多,我不知道该关注哪一篇以及如何关注 (顺便说一句,python版本3.8.1,torch 2.0.0,vision 0.15.1,cuda 11.8,没有安装cudatoolkit,因为我不知道选择哪一个

python detectron
1个回答
0
投票

您可以使用“conda install pytorch torchvision cudatoolkit=10.1 -c pytorch”来下载带有pytorch和torchvision的cuda工具包。一旦你有了这个,我就找到了一个有用的资源,让我开始使用 detectorron2,这里是 google colab 页面:https://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5#scrollTo=7unkuuiqLdqd

© www.soinside.com 2019 - 2024. All rights reserved.