diff --git a/facefusion/choices.py b/facefusion/choices.py index 5b3e8ae..cda573f 100755 --- a/facefusion/choices.py +++ b/facefusion/choices.py @@ -46,6 +46,7 @@ image_type_set : ImageTypeSet =\ 'bmp': 'image/bmp', 'jpeg': 'image/jpeg', 'png': 'image/png', + 'tiff': 'image/tiff', 'webp': 'image/webp' } video_type_set : VideoTypeSet =\ @@ -59,7 +60,7 @@ video_type_set : VideoTypeSet =\ audio_formats : List[AudioFormat] = list(audio_type_set.keys()) image_formats : List[ImageFormat] = list(image_type_set.keys()) video_formats : List[VideoFormat] = list(video_type_set.keys()) -temp_frame_formats : List[ImageFormat] = [ 'bmp', 'jpeg', 'png' ] +temp_frame_formats : List[ImageFormat] = [ 'bmp', 'jpeg', 'png', 'tiff' ] output_audio_encoders : List[AudioEncoder] = [ 'aac', 'libmp3lame', 'libopus', 'libvorbis' ] output_video_encoders : List[VideoEncoder] = [ 'libx264', 'libx265', 'libvpx-vp9', 'h264_nvenc', 'hevc_nvenc', 'h264_amf', 'hevc_amf', 'h264_qsv', 'hevc_qsv', 'h264_videotoolbox', 'hevc_videotoolbox' ] diff --git a/facefusion/filesystem.py b/facefusion/filesystem.py index f40c012..aacdba2 100644 --- a/facefusion/filesystem.py +++ b/facefusion/filesystem.py @@ -34,6 +34,8 @@ def get_file_format(file_path : str) -> Optional[str]: if file_extension: if file_extension == '.jpg': return 'jpeg' + if file_extension == 'tif': + return 'tiff' return file_extension.lower().lstrip('.') return None diff --git a/facefusion/typing.py b/facefusion/typing.py index b08898f..9bd7221 100755 --- a/facefusion/typing.py +++ b/facefusion/typing.py @@ -110,9 +110,9 @@ FaceMaskRegion = Literal['skin', 'left-eyebrow', 'right-eyebrow', 'left-eye', 'r FaceMaskRegionSet = Dict[FaceMaskRegion, int] AudioFormat = Literal['mp3', 'ogg', 'wav'] -ImageFormat = Literal['bmp', 'jpeg', 'png', 'webp'] +ImageFormat = Literal['bmp', 'jpeg', 'png', 'tiff', 'webp'] VideoFormat = Literal['avi', 'mkv', 'mov', 'mp4', 'webm'] -TempFrameFormat = Literal['bmp', 'jpeg', 'png'] +TempFrameFormat = Literal['bmp', 'jpeg', 'png', 'tiff'] AudioTypeSet = Dict[AudioFormat, str] ImageTypeSet = Dict[ImageFormat, str] VideoTypeSet = Dict[VideoFormat, str]