python: 使用os.system为文件批量建立软连接

前言

本篇记录一下使用python在ubuntu中给文件批量建立软链接的方法。

批量软链接的原因

因为训练时的数据集存放的位置路径比较长，想直接放在训练工程下，直接复制粘贴会占据大量空间；而这个数据集也很常用，其它算法也会用到它，所以剪切到该工程文件夹也不合适。

于是在本文件夹下建立数据集的软链接是最好的选择，但是我的数据集读取代码要求了数据集的排列格式，于是建立软链接时，还得按照格式要求分别为每个数据文件夹建立一个软链接。

os.system()

python的os模块提供了os.system()方法，可以执行系统命令。于是可以通过该方法进行ln软链接操作，而批量ln就变成做循环了。

代码

import os

root_path = os.path.abspath('~/Datasets/mydataset')
target_path = os.path.abspath('./')
source_dir = os.path.join(root_path, 'trainvaltest')
target_dir = os.path.join(target_path, 'dataset')
for f in os.listdir(source_dir):
    for ff in os.listdir(os.path.join(source_dir, f)):
        source = os.path.join(source_dir, f, ff)
        target = os.path.join(target_dir, ff)
        if os.path.isdir(source):
            print(source)
            os.system('ln -s {} {}'.format(source, target))