当前位置：首页 > news >正文

复现------

news 2025/11/12 8:15:34

配置一个项目环境

最好在终端，进入到项目目录 cd 项目目录如cd ～/code/MOTR

然后配环境

安装pytorch 和torchvision需要考虑匹配的问题，上官网

https://pytorch.org/get-started/previous-versions/找历史版本对应的cuda版本的torch和torchvision ctrl+f搜索，conda不好装用匹配装用pip装，这个只是装对应cuda版本的torch和torchvision，并没有装cuda

安装cuda，从cuda官网装

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=runfile_local

切换cuda版本

https://blog.csdn.net/qq_41917697/article/details/114437924

cannot import name ‘__NewEmptyTensorOp‘

权重一般在项目下新建一个checkpoints文件夹,把.pth权重文件放在checkpoints文件夹

pycharm里对于参数的配置，弄好配置以后，先应用再确定

比如跑demo.py,编辑配置

官网给的是

EXP_DIR=exps/e2e_motr_r50_joint
python3 demo.py \--meta_arch motr \--dataset_file e2e_joint \--epoch 200 \--with_box_refine \--lr_drop 100 \--lr 2e-4 \--lr_backbone 2e-5 \--pretrained ${EXP_DIR}/motr_final.pth \--output_dir ${EXP_DIR} \--batch_size 1 \--sample_mode 'random_interval' \--sample_interval 10 \--sampler_steps 50 90 120 \--sampler_lengths 2 3 4 5 \--update_query_pos \--merger_dropout 0 \--dropout 0 \--random_drop 0.1 \--fp_ratio 0.3 \--query_interaction_layer 'QIM' \--extra_track_attn \--resume ${EXP_DIR}/motr_final.pth \--input_video figs/demo.avi

script下边那一栏填写如下改了EXP_DIR（相当于全局变量）、pretrained(模型权重存放位置)、 output_dir（results是默认自己会创建的） ‘QIM’->QIM、resume

--meta_arch
motr
--dataset_file
e2e_joint
--epoch
200
--with_box_refine
--lr_drop
100
--lr
2e-4
--lr_backbone
2e-5
--pretrained
/home/dby/code/MOTR/checkpoints/model_motr_final.pth
--output_dir
result/
--batch_size
1
--sample_mode
'random_interval'
--sample_interval
10
--sampler_steps
50
90
120
--sampler_lengths
2
3
4
5
--update_query_pos
--merger_dropout
0
--dropout
0
--random_drop
0.1
--fp_ratio
0.3
--query_interaction_layer
QIM
--extra_track_attn
--resume
/home/dby/code/MOTR/checkpoints/model_motr_final.pth
--input_video
figs/demo.avi