0

我有一个图像文件,我想检查它是否是使用 python 的图像序列的一部分。

例如我从这个文件开始:

/projects/image_0001.jpg

我想检查文件是否是序列的一部分,即

/projects/image_0001.jpg
/projects/image_0002.jpg
/projects/image_0003.jpg
...

如果我可以确定文件名是否可以是序列的艺术,即是否存在文件名的数字序列,则检查是否存在图像序列似乎很简单

我的第一个问题是要求用户将####数字添加到文件路径中,并输入开始和结束帧号来替换散列,但这显然不是很用户友好。有没有办法用正则表达式或类似的东西检查字符串中的数字序列?

4

2 回答 2

3

我假设问题更多是因为能够区分磁盘上的排序文件,而不是知道有关文件名本身的任何特定信息。

如果是这样的话,并且您正在寻找的是足够聪明的东西,可以列出如下列表:

  • /path/to/file_1.png
  • /path/to/file_2.png
  • /path/to/file_3.png
  • ...
  • /path/to/file_10.png
  • /path/to/image_1.png
  • /path/to/image_2.png
  • ...
  • /path/to/image_10.png

并返回一个结果说 - 我有 2 个文件序列: /path/to/file_#.png 和 /path/to/image_#.png 你将需要 2 次通过 - 第一次通过来确定文件的有效表达式,第二遍找出所有其他文件满足该要求。

您还需要知道是否要支持间隙(是否需要连续)

  • /path/to/file_1.png
  • /path/to/file_2.png
  • /path/to/file_3.png
  • /path/to/file_5.png
  • /path/to/file_6.png
  • /path/to/file_7.png

这是 1 个序列 (/path/to/file_#.png) 还是 2 个序列 (/path/to/file_1-3.png, /path/to/file_5-7.png)

另外 - 你想如何处理序列中的数字文件?

  • /path/to/file2_1.png
  • /path/to/file2_2.png
  • /path/to/file2_3.png

等等

考虑到这一点,这就是我将如何完成它:

    import os.path
    import projex.sorting
    import re

    def find_sequences( filenames ):
        """
        Parse a list of filenames into a dictionary of sequences.  Filenames not
        part of a sequence are returned in the None key

        :param      filenames | [<str>, ..]

        :return     {<str> sequence: [<str> filename, ..], ..}
        """
        local_filenames   = filenames[:]
        sequence_patterns = {}
        sequences         = {None: []}

        # sort the files (by natural order) so we always generate a pattern
        # based on the first potential file in a sequence
        local_filenames.sort(projex.sorting.natural)

        # create the expression to determine if a sequence is possible
        # we are going to assume that its always going to be the 
        # last set of digits that makes a sequence, i.e.
        #
        #    test2_1.png
        #    test2_2.png
        #
        # test2 will be treated as part of the name
        # 
        #    test1.png
        #    test2.png
        #
        # whereas here the 1 and 2 are part of the sequence
        #
        # more advanced expressions would be needed to support
        # 
        #    test_01_2.png
        #    test_02_2.png
        #    test_03_2.png

        pattern_expr = re.compile('^(.*)(\d+)([^\d]*)$')

        # process the inputed files for sequences
        for filename in filenames:
            # first, check to see if this filename matches a sequence
            found = False
            for key, pattern in sequence_patterns.items():
                match = pattern.match(filename)
                if ( not match ):
                    continue

                sequences[key].append(filename)
                found = True
                break

            # if we've already been matched, then continue on
            if ( found ):
                continue

            # next, see if this filename should start a new sequence
            basename      = os.path.basename(filename)
            pattern_match = pattern_expr.match(basename)
            if ( pattern_match ):
                opts = (pattern_match.group(1), pattern_match.group(3))
                key  = '%s#%s' % opts

                # create a new pattern based on the filename
                sequence_pattern = re.compile('^%s\d+%s$' % opts)

                sequence_patterns[key] = sequence_pattern
                sequences[key] = [filename]
                continue

            # otherwise, add it to the list of non-sequences
            sequences[None].append(filename)

        # now that we have grouped everything, we'll merge back filenames
        # that were potential sequences, but only contain a single file to the
        # non-sequential list
        for key, filenames in sequences.items():
            if ( key is None or len(filenames) > 1 ):
                continue

            sequences.pop(key)
            sequences[None] += filenames

        return sequences

以及一个示例用法:

>>> test =   ['test1.png','test2.png','test3.png','test4.png','test2_1.png','test2_2.png','test2_3.png','test2_4.png']
>>> results = find_sequences(test)
>>> results.keys()
[None, 'test#.png', 'test2_#.png']

那里有一种方法是指自然排序,这是一个单独的主题。我刚刚使用了我的 projex 库中的自然排序方法。它是开源的,所以如果您想使用或查看它,请访问:http: //dev.projexsoftware.com/projects/projex

但是该主题已在论坛的其他地方有所涉及,因此仅使用了图书馆中的方法。

于 2012-08-08T01:34:28.417 回答
2

使用 python 的re模块来查看字符串是否包含数字序列相对容易。你可以这样做:

mo = re.findall('\d+', filename)

这将返回所有数字序列的列表filename。如果:

  • 有一个结果(即文件名仅包含一个数字序列),并且
  • 随后的文件名具有相同长度的单个数字序列,并且
  • 第二个数字序列比前一个大 1

...那么也许它们是序列的一部分。

于 2012-08-08T00:39:37.217 回答