安装RF-DETR需要按照以下步骤进行:
- Preparação ambiental:确保安装Python 3.9+和PyTorch 1.13.0+版本。若使用GPU,需提前通过
nvidia-smi
命令验证驱动状态 - Instalando o PyTorch: Implementação
pip install torch>=1.13.0 torchvision>=0.14.0
Instalação de dependências básicas - Código de download:通过git命令克隆仓库
git clone https://github.com/roboflow/rf-detr.git
e insira o catálogo - Instalação de dependências: Executar
pip install rfdetr
自动安装numpy、supervision等必要库
验证安装是否成功可执行以下Python代码:
from rfdetr import RFDETRBase
print("安装成功")
若无报错则表明环境配置正确。整个过程约需5-10分钟,具体时间取决于网络速度和硬件配置。
Essa resposta foi extraída do artigoRF-DETR: um modelo de código aberto para detecção de objetos visuais em tempo realO