Ubuntu下HoiTransformer复现pytorch环境搭建+解决问题

一、准备工作

论文：End-to-End Human Object Interaction Detection with HOI Transformer

代码：GitHub - bbepoch/HoiTransformer: This is the code for HOI Transformer 

二、环境搭建

硬件/系统：服务器GPU3090，Ubuntu20.。。（忘记了），CUDA11.2

虚拟环境：python3.6，torch1.10.0+cu113，torchaudio0.10.0+cu113，torchvision0.11.1+cu113  

虚拟环境创建过程略

三、复现过程

1.克隆代码（github难以直接克隆于是去gitee克隆了，最好克隆不要直接下载压缩包）

git clone https://gitee.com/hoi_xd/HoiTransformer

2.下载预训练模型

cd data/detr_coco && bash download_model.sh

3.下载注释文件

<