【计算机视觉 | 目标检测】RegionCLIP的环境部署和代码测试（含源代码）

阳光穿透心脏的1/2处 2023-10-13 17:31 24阅读 0赞

#### 文章目录 ####

*  一、导读
 *  二、环境部署
 *  三、权重文件配置
 *  四、数据集的配置
 *  五、代码执行

## 一、导读 ##

RegionCLIP: Region-based Language-Image Pretraining

This is the official PyTorch implementation of RegionCLIP (CVPR 2022).

![在这里插入图片描述][1a9eafc6106e42b38b0cbbf82a5fd35c.png]

我们提出 RegionCLIP，它显著扩展了 CLIP 以学习区域级视觉表示。 RegionCLIP 支持图像区域和文本概念之间的细粒度对齐，从而支持基于区域的推理任务，包括零样本对象检测和开放词汇对象检测。

*  预训练：我们利用 CLIP 模型将图像区域与模板标题进行匹配，然后预训练我们的模型以对齐这些区域-文本对。
 *  零样本推理：经过预训练后，学习到的区域表示支持用于对象检测的零样本推理。
 *  学习到的 RegionCLIP 模型可以通过附加的对象检测注释进行进一步微调，从而使我们的模型能够用于完全监督或开放词汇对象检测。
 *  我们的方法展示了零样本目标检测和开放词汇目标检测的最先进结果。

## 二、环境部署 ##

下一步，下载RegionCLIP文件，下载地址作者已经给出：

https://github.com/microsoft/RegionCLIP.git

![在这里插入图片描述][42578a5a250546218915a1c337502dd1.png]  
完成RegionCLIP文件的下载后，便需要配置RegionCLIP进行zero-shot推理的环境配置以及所需要的配置文件。

然后进行环境配置：

!python -m pip install -e RegionCLIP

![在这里插入图片描述][2b58b330db3d40db96110366ce7eef53.png]  
![在这里插入图片描述][92695fea47a243dfa41a19df33a4cf9f.png]

安装其他所需的文件：

!pip install opencv-python timm diffdist h5py sklearn ftfy

![在这里插入图片描述][cb4f16b399264d0e873d697ffd4f9a75.png]

!pip install git+https://github.com/lvis-dataset/lvis-api.git

![在这里插入图片描述][e9854f103c73479baeb156b1f81c822c.png]

## 三、权重文件配置 ##

首先需要下载一个pretrained\_ckpt文件夹下到RegionCLIP文件夹下：

![在这里插入图片描述][f386bcdaaa3145cc92f7ed1047faf009.png]  
文件夹的地址为：

https://drive.google.com/drive/folders/1hzrJBvcCrahoRcqJRqzkIGFO_HUSJIii

![在这里插入图片描述][d2ae1ee3c87c46d7b844eb56ba7df1a6.png]

由于共享的文件夹不是pretrained\_ckpt名称，所以将其重命名为pretrained\_ckpt即可。

## 四、数据集的配置 ##

还需要一个lvis验证集的label数据集，这个直接去官网下载即可，不是很大，官网地址：

![在这里插入图片描述][30a050240b804aaa8c7473fb3bec3a96.png]

注意的是，要选择路径”RegionCLIP””datasets”后再点击新建，把文件夹上传到RegionCLIP/datasets下，并且文件夹名为lvis：

![在这里插入图片描述][75e23ecd77be4d3fb1b7818e9909eeb7.png]

至此，你已经完成好了RegionCLIP的zero-shot测试的相关文件配置，下面要做的就是running code啦。

具体的一些配置如下：

![在这里插入图片描述][86498a0a6fa94d2b8e5a772ba046312d.png]

## 五、代码执行 ##

记得先切换到对应的目录下：![在这里插入图片描述][0f4e8a829e684bb9a9d5063240916ab6.png]

执行如下代码：

python ./tools/train_net.py \
    --eval-only \
    --num-gpus 1 \
    --config-file ./configs/LVISv1-InstanceSegmentation/CLIP_fast_rcnn_R_50_C4_custom_img.yaml \
    MODEL.WEIGHTS ./pretrained_ckpt/regionclip/regionclip_pretrained-cc_rn50x4.pth \
    MODEL.CLIP.TEXT_EMB_PATH ./pretrained_ckpt/concept_emb/lvis_1203_cls_emb_rn50x4.pth \
    MODEL.CLIP.OFFLINE_RPN_CONFIG ./configs/LVISv1-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml \
    MODEL.CLIP.TEXT_EMB_DIM 640 \
    MODEL.RESNETS.DEPTH 200 \
    MODEL.ROI_BOX_HEAD.POOLER_RESOLUTION 18 \

![在这里插入图片描述][64e5be1e99f04752bd763311011021ea.png]

运行结果如下：

![在这里插入图片描述][c06562ad485b4343921b9af18c7bc59e.png]  
然后会在RegionCLIP目录生成一个“output/inference/lvis\_instances\_results.json"文件夹：

![在这里插入图片描述][9c265cb9922a4cd0bd4f18f47aded5ac.png]

为了可视化最后的zero-shot测试结果，执行如下代码：

python ./tools/visualize_json_results.py \
    --input ./output/inference/lvis_instances_results.json \
    --output ./output/regions \
    --dataset lvis_v1_val_custom_img \
    --conf-threshold 0.05 \
    --show-unique-boxes \
    --max-boxes 25 \
    --small-region-px 8100\

然后找到根据路径RegionCLIP/output/regions找到对应的测试结果图片：

![在这里插入图片描述][81dfd24d076f4334a82b44750452ddc6.png]

结果展示如下：

![在这里插入图片描述][7b806195b83e48c18040d6b8e6c72937.png]

![在这里插入图片描述][a518aadadbfb4d4e8dfbe8a9fa08078d.png]

从测试结果可以看出，RegionCLIP的zero-shot推理主要集中在未知类别的目标检测，也就是说对于一些常见的类别如person、book、cup等，RegionCLIP的zero-shot就不进行定位和识别，只是检测少见的类，也就是一般数据集如coco里面没有的类别，比如第一张图片里的pantyose（连裤袜，这里识别错了）、kimono（和服）、以及第二张图片里的shopping\_cart、short\_pants、deck\_chair等这些类别在常见的数据集是没有标记的，只是一些关于图片的描述中可能会出现的，也就是image-caption数据集中才会有的。

[1a9eafc6106e42b38b0cbbf82a5fd35c.png]: https://img-blog.csdnimg.cn/1a9eafc6106e42b38b0cbbf82a5fd35c.png
[42578a5a250546218915a1c337502dd1.png]: https://img-blog.csdnimg.cn/42578a5a250546218915a1c337502dd1.png
[2b58b330db3d40db96110366ce7eef53.png]: https://img-blog.csdnimg.cn/2b58b330db3d40db96110366ce7eef53.png
[92695fea47a243dfa41a19df33a4cf9f.png]: https://img-blog.csdnimg.cn/92695fea47a243dfa41a19df33a4cf9f.png
[cb4f16b399264d0e873d697ffd4f9a75.png]: https://img-blog.csdnimg.cn/cb4f16b399264d0e873d697ffd4f9a75.png
[e9854f103c73479baeb156b1f81c822c.png]: https://img-blog.csdnimg.cn/e9854f103c73479baeb156b1f81c822c.png
[f386bcdaaa3145cc92f7ed1047faf009.png]: https://img-blog.csdnimg.cn/f386bcdaaa3145cc92f7ed1047faf009.png
[d2ae1ee3c87c46d7b844eb56ba7df1a6.png]: https://img-blog.csdnimg.cn/d2ae1ee3c87c46d7b844eb56ba7df1a6.png
[30a050240b804aaa8c7473fb3bec3a96.png]: https://img-blog.csdnimg.cn/30a050240b804aaa8c7473fb3bec3a96.png
[75e23ecd77be4d3fb1b7818e9909eeb7.png]: https://img-blog.csdnimg.cn/75e23ecd77be4d3fb1b7818e9909eeb7.png
[86498a0a6fa94d2b8e5a772ba046312d.png]: https://img-blog.csdnimg.cn/86498a0a6fa94d2b8e5a772ba046312d.png
[0f4e8a829e684bb9a9d5063240916ab6.png]: https://img-blog.csdnimg.cn/0f4e8a829e684bb9a9d5063240916ab6.png
[64e5be1e99f04752bd763311011021ea.png]: https://img-blog.csdnimg.cn/64e5be1e99f04752bd763311011021ea.png
[c06562ad485b4343921b9af18c7bc59e.png]: https://img-blog.csdnimg.cn/c06562ad485b4343921b9af18c7bc59e.png
[9c265cb9922a4cd0bd4f18f47aded5ac.png]: https://img-blog.csdnimg.cn/9c265cb9922a4cd0bd4f18f47aded5ac.png
[81dfd24d076f4334a82b44750452ddc6.png]: https://img-blog.csdnimg.cn/81dfd24d076f4334a82b44750452ddc6.png
[7b806195b83e48c18040d6b8e6c72937.png]: https://img-blog.csdnimg.cn/7b806195b83e48c18040d6b8e6c72937.png
[a518aadadbfb4d4e8dfbe8a9fa08078d.png]: https://img-blog.csdnimg.cn/a518aadadbfb4d4e8dfbe8a9fa08078d.png