【ava 数据集】训练集和验证集之间的比例

这里简单说一下关于ava数据集的训练集和验证集之间的比例

在ava数据集中，训练视频235个，验证视频64个，比例为100:28，也就是说，我们在自己做数据集的时候，尽量按照这个比例来制作

下面是代码部分：

import csv

csv_file=csv.reader(open('ava_train_v2.2.csv','r'))
next(csv_file)

name = ''
count = 0
for line in csv_file:
    tempname=name
    name = line[0]
    if name != tempname:
        count= count + 1

print("train:",count)

csv_file=csv.reader(open('ava_val_v2.2.csv','r'))
next(csv_file)

name = ''
count = 0
for line in csv_file:
    tempname=name
    name = line[0]
    if name != tempname:
        count= count + 1

print("val:",count)