[视频目标检测]:使用MEGA

您所在的位置:网站首页 dff文件夹 [视频目标检测]:使用MEGA

[视频目标检测]:使用MEGA

2024-04-07 15:43| 来源: 网络整理| 查看: 265

1.创建环境创建虚拟环境conda create --name MEGA -y python=3.7 source activate MEGA安装基础包conda install ipython pip pip install ninja yacs cython matplotlib tqdm opencv-python scipy export INSTALL_DIR=$PWD

安装pytorch

在安装pytorch的时候,原作者是这样的:

conda install pytorch=1.3.0 torchvision cudatoolkit=10.0 -c pytorch

但实际上使用cuda11.0+pytorch1.7也可以编译跑通,所以在这一步我们将其替换成:

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch然后就是作者使用到的coco数据集和cityperson数据集的安装:cd $INSTALL_DIR git clone https://github.com/cocodataset/cocoapi.git cd cocoapi/PythonAPId python setup.py build_ext install cd $INSTALL_DIR git clone https://github.com/mcordts/cityscapesScripts.git cd cityscapesScripts/ python setup.py build_ext install安装apex:(可省略) (建议省略,没省略运行报错)git clone https://github.com/NVIDIA/apex.git cd apex python setup.py install --cuda_ext --cpp_ext

如果使用的是cuda11.0+pytorch1.7这里会报错

deform_conv_cuda.cu(155): error: identifier "AT_CHECK " is undefined

解决:在mega_core/csrc/cuda/deform_conv_cuda.cu 和 mega_core/csrc/cuda/deform_pool_cuda.cu文件的开头加上如下代码:

#ifndef AT_CHECK #define AT_CHECK TORCH_CHECK #endif

实际上原作者并没有使用到apex来进行混合精度训练,这一步也可省略,若省略的话在代码中需要修改几处地方:首先是mega_core/engine/trainer.py中的开头导入apex包注释掉,

108-109行改为:

losses.backward()

还有tools/train_net.py中33行-36行注释掉

# try: # from apex import amp # except ImportError: # raise ImportError('Use APEX for multi-precision via apex.amp')

50行也注释掉:

#model, optimizer = amp.initialize(model, optimizer, opt_level=amp_opt_level)

还有mega_core/layers/nms.py,注释掉第5行

第8行改为:

nms = _C.nms

还有mega_core/layers/roi_align.py注释掉第10、57行

还有mega_core/layers/roi_pool.py注释掉第10、56行

这样应该就可以了。

2.下载和初始化mega.pytorch# install PyTorch Detection cd $INSTALL_DIR git clone https://github.com/Scalsol/mega.pytorch.git cd mega.pytorch # the following will install the lib with # symbolic links, so that you can modify # the files if you want and won't need to # re-build it python setup.py build develop pip install 'pillow


【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3