FCOS目标检测

本文档详细介绍了FCOS目标检测模型的环境配置,包括CUDA 10.0、Python 3.6.15、PyTorch 1.4.0和torchvision 0.2.1等依赖。同时,阐述了如何下载并编译相关文件,如cocoapi和FCOS项目。在编译过程中,若遇到问题,建议使用匹配版本的工具。测试阶段,通过加载官方.pth文件验证模型,参考了相关博客解决遇到的问题。
摘要由CSDN通过智能技术生成

环境:

cuda版本:Cuda compilation tools, release 10.0, V10.0.130

python:3.6.15

torch:1.4.0

torchvision:0.2.1

ninja:1.10.2

conda create -n FCOS python=3.6
conda activate FCOS

conda install ipython
pip install ninja==1.10.2 yacs cython matplotlib tqdm

pip install torch==1.4.0 torchvision==0.2.1 -i https://pypi.douban.com/simple

下载相关文件:

把下载的文件放在自己想放的地方就行,如我放在downloads下:cd home/Downloads

cocoapi:

git clone https://github.com/cocodataset/cocoapi.git

FCOS:

git clone https://github.com/tianzhi0549/FCOS.git

下面进行一系列编译:

cd home/Downloads/cocoapi/PythonAPI
python setup.py build_ext install

cd home/Downloads/FCOS
python setup.py build develop --no-deps

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值