fatf.utils.data.occlusion.Occlusion.occlude_segments_vectorised removes a dimension for input arrays of shape (1, n)

[So-Cool]

Please check that you have already viewed existing issues and that the bug you are reporting is new.

Python version

• 2.x
• 3.6.x
• 3.7.11

Package versions

• fat-forensics: 0.1.1
• numpy: 1.17.3
• scipy: 1.4.1

Description

fatf.utils.data.occlusion.Occlusion.occlude_segments_vectorised does not differentiate between an input array of shape (1, n) and (n, ). This is caused by line 648.

Source Code

import fatf.utils.data.occlusion as fatf_occlusion
occluder = fatf_occlusion.Occlusion()
occluder.occlude_segments_vectorised([[1, 1, 1]])
# The output should be of shape (1, ...)