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

最新推荐文章于 2025-04-28 09:23:14 发布

原创最新推荐文章于 2025-04-28 09:23:14 发布 · 1.8k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python #ava #ava数据集 #训练集 #验证集

本文介绍了AVA数据集的训练集和验证集的比例，并提供了一段用于统计训练集和验证集视频数量的Python代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这里简单说一下关于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)