利用现有模型处理面部视频获取特征向量(1)

第一步：挑选合适的模型

目前有两个比较符合心意的模型，一个是video-features另外一个是STPN

让我来检验一下，哪一个比较好

STPN

在按照readme安装环境和配置的时候，发现mmpycocotools死活报错，重装Cython至0.29.33版本后就可以装上mmpycocotools了。

以及dataset的链接失效了，需要在官网下ILSVRC2015。不过我不知道该下载哪个分类下的dataset。

我下了上图中的三个（属实是乱下） ，但是下面这个VID的链接失效了

于是我直接放弃，尝试下面一个方案了

video-features

# clone the repo and change the working directory
git clone https://github.com/v-iashin/video_features.git
cd video_features# install environment
conda env create -f conda_env.ymlconda activate video_features

readme安装完环境之后，可以使用以下代码检测pytorch是否支持gpu

python
import torch
print(torch.cuda.is_available())

我的输出是true，但是为什么程序运行之后会被killed？

conda activate video_features
python main.py \feature_type=r21d \device="cuda:0" \video_paths="[/mnt/c/Users/DDDCY/Desktop/short-mp4/0.mp4,/mnt/c/Users/DDDCY/Desktop/short-mp4/1.mp4,/mnt/c/Users/DDDCY/Desktop/short-mp4/2.mp4,/mnt/c/Users/DDDCY/Desktop/short-mp4/3.mp4,/mnt/c/Users/DDDCY/Desktop/short-mp4/4.mp4,/mnt/c/Users/DDDCY/Desktop/short-mp4/5.mp4,/mnt/c/Users/DDDCY/Desktop/short-mp4/6.mp4]"

我还只是试个水，只放了七个视频呢，没有把所有69个视频放进去（我放的还是短视频，一个30秒，原视频每个90秒呢！！！） 

再看一下被killed的原因

(video_features) dddcyy@dddcyy6100846:~/video_features$ sudo dmesg | tail -7
[sudo] password for dddcyy:
[118484.930910] [  10029]  1000 10029     5063      519    77824        0           100 systemd
[118484.930912] [  10030]  1000 10030     5287       67    77824      365           100 (sd-pam)
[118484.930919] [  10045]  1000 10045     1518      392    57344        0             0 bash
[118484.930926] [  10120]  1000 10120 16381865  3898516 41156608  1046686             0 python
[118484.930928] oom-kill:constraint=CONSTRAINT_NONE,nodemask=(null),cpuset=/,mems_allowed=0,global_oom,task_memcg=/init.scope,task=python,pid=10120,uid=1000
[118484.931253] Out of memory: Killed process 10120 (python) total-vm:65527460kB, anon-rss:15519112kB, file-rss:0kB, shmem-rss:74952kB, pgtables:40192kB oom_sc

但是我不确定是不是只是因为gpu的内存不够导致的，还是说还有其他的原因。让我明天在服务器上面试一试就知道了