资讯动态

IM-CMDet代码复现(Ubuntu (Linux))​

发布时间:2026/8/22 13:14:48 来源:尧图企业网站定制
资源准备代码百度网盘 请输入提取码数据集百度网盘 请输入提取码创建并激活环境conda create -n imcmdet python3.9 -y conda activate imcmdet安装 PyTorchpip install torch1.13.1cu116 torchvision0.14.1cu116 torchaudio0.13.1 --index-url https://download.pytorch.org/whl/cu116安装 mmcv-full 1.7.2pip install mmcv-full1.7.2 -f https://download.openmmlab.com/mmcv/dist/cu116/torch1.13/index.html安装其他依赖pip install scipy terminaltables pycocotools matplotlib seaborn scikit-learn tensorboard opencv-python修改数据集位置修改代码里面数据集的位置改成自己数据集的位置data_root /home/buu622/桌面/RGBTDronePerson/设置PYTHONPATHIM_CMDet/tools/train.py的开头加上如下的代码import sys import os.path as osp sys.path.insert(0, osp.join(osp.dirname(__file__), ..))下载预训练权重wget -4 -P /home/buu622/.cache/torch/hub/checkpoints/ https://download.pytorch.org/models/resnet50-0676ba61.pthNumpy降级pip install numpy2开始训练python tools/train.py data/IM_CMDet_configs/IMCMDet_res50_DSJE_1x_droneperson.py --work-dir work_dirs/paper_reproduce

读完文章,也想定制专属网站?

尧图设计师 24 小时内与您沟通定制方案

免费获取报价