我根据视频分类之UCF-101上的CNN方法详解 - 知乎 (zhihu.com)介绍的方法下载了源代码和数据集,第一个py是移动文件代码。按此链接要求:
把数据集文件放在了data下。
其中ucfTrainTestlist文件夹是这样的:
数据集文件夹是这样的:
源代码中,我把两个文件路径改了:
"""
After extracting the RAR, we run this to move all the files into
the appropriate train/test folders.
Should only run this file once!
"""
import os
import os.path
def get_train_test_lists(version='01'):
"""
Using one of the train/test files (01, 02, or 03), get the filename
breakdowns we'll later use to move everything.
"""
# Get our files based on version.
test_file = 'E:/Graduation Project/UCF-101_video_classification-master/data/ucfTrainTestlist/testlist' + version + '.txt'
train_file = 'E:/Graduation Project/UCF-101_video_classification-master/data/ucfTrainTestlist/trainlist' + version + '.txt'
# Build the test list.
with open(test_file) as fin:
test_list = [row.strip() for row in list(fin)]
# Build the train list. Extra step to remove the class index.
with open(train_file) as fin:
train_list = [row.strip() for row in list(fin)]
train_list = [row.split(' ')[0] for row in train_list]
# Set the groups in a dictionary.
file_groups = {
'train': train_list,
'test': test_list
}
return file_groups
def move_files(file_groups):
"""This assumes all of our files are currently in _this_ directory.
So move them to the appropriate spot. Only needs to happen once.
"""
# Do each of our groups.
for group, videos in file_groups.items():
# Do each of our videos.
for video in videos:
# Get the parts.
parts = video.split('/')
classname = parts[0]
filename = parts[1]
# Check if this class exists.
if not os.path.exists(group + '/' + classname):
print("Creating folder for %s/%s" % (group, classname))
os.makedirs(group + '/' + classname)
# Check if we have already moved this file, or at least that it
# exists to move.
if not os.path.exists(filename):
print("Can't find %s to move. Skipping." % (filename))
continue
# Move it.
dest = group + '/' + classname + '/' + filename
print("Moving %s to %s" % (filename, dest))
os.rename(filename, dest)
print("Done.")
def main():
"""
Go through each of our train/test text files and move the videos
to the right place.
"""
# Get the videos in groups so we can move them.
group_lists = get_train_test_lists()
# Move the files.
move_files(group_lists)
if __name__ == '__main__':
main()
但是运行会出现这样:
请问这是为啥?