第一步:挑选合适的模型
目前有两个比较符合心意的模型,一个是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的内存不够导致的,还是说还有其他的原因。让我明天在服务器上面试一试就知道了