使用colab进行REDS数据集整合时,运行py文件单元格无输出

图超分小白一枚,最近在看代码 ,想着自己拿来复现一下,在准备数据集阶段选用了文中给的REDS数据集,并且下载解压到我的文件夹里了,我已经把下面的代码修改成了我保存的路径,但是不知道为什么colab每次一点运行就没有反应,转圈一分钟后自动结束。

以下是整合数据集的原代码

import glob
import os


def regroup_reds_dataset(train_path, val_path):
    """Regroup original REDS datasets.

    We merge train and validation data into one folder, and separate the
    validation clips in reds_dataset.py.
    There are 240 training clips (starting from 0 to 239),
    so we name the validation clip index starting from 240 to 269 (total 30
    validation clips).

    Args:
        train_path (str): Path to the train folder.
        val_path (str): Path to the validation folder.
    """
    # move the validation data to the train folder
    val_folders = glob.glob(os.path.join(val_path, '*'))
    for folder in val_folders:
        new_folder_idx = int(folder.split('/')[-1]) + 240
        os.system(f'cp -r {folder} {os.path.join(train_path, str(new_folder_idx))}') #这里会不会有一些问题呢?


if __name__ == '__main__':
    # train_sharp
    train_path = 'datasets/REDS/train_sharp'
    val_path = 'datasets/REDS/val_sharp'
    regroup_reds_dataset(train_path, val_path)

    # train_sharp_bicubic
    train_path = 'datasets/REDS/train_sharp_bicubic/X4'
    val_path = 'datasets/REDS/val_sharp_bicubic/X4'
    regroup_reds_dataset(train_path, val_path)


这里是现象

img

我猜测是调用os.system这里可能出了问题,因为colab中使用指令和pycharm不同,我试着在os.systerm里加!不是很行,希望能有大佬帮忙看一下,不尽感激!!!

感觉应该是没问题的,这个脚本本身也没有输出语句,就只是拷贝一下文件,所以运行完如果文件拷贝成功了就没问题的,如果没有拷贝成功的话可能是路径有问题