-
-
Notifications
You must be signed in to change notification settings - Fork 6.9k
Fix square brackets in the target path #532
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
|
There is glob.escape() for that. Please open another PR. |
|
the proper way to use glob.glob("*.png", root_dir=output_dir) |
It returns only list of filenames, without trailing paths. Or am I doing something wrong (I'm a still a newcomer in Python)? |
|
if you want you can do it's one additional step but easier to understand and safer than writing complex regex |
|
I closed that PR initially because a Regex like this is a nono ... I agree with @phineas-pta |
I don't see anything wrong with regular expressions, but you're the boss here 🙃. I've come up with an alternative approach without using regex, take a look on it. |
|
short regex are sweet but when they become long complex, code get harder to maintain/debug for e.g. you tried to fix problem with |
|
@henryruhs actually, you were right: |
It is not a Windows problem; it is the behavior of |
|
I merged it... chatgpt told me it can be simplified to: as of better readability. |
|
@pozitronik @henryruhs chatgpt's answer fails, please don't blindly believe it 😂 |
* ffmpeg platform-agnostic hardware-acceleration
* clear CUDA cache after swapping
on low VRAM + ffmpeg cuda acceleration, clearing cache prevent cuda out-of-memory error
* check torch gpu before clearing cache
* torch check nvidia only
* syntax error
* Adjust comment
* Normalize ARGS
* Remove path normalization
* Remove args overrides
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Revert to Ubuntu test only as Windows hangs
* Simplified the way to maintain aspect ratio of the preview, and maintaining aspect ratio of the miniatures
* Change face and target images from contain to fit
* Improve status output
* Massive utilities and core refactoring
* Fix sound
* Fix sound part2
* Fix more
* Move every UI related thing to ui.py
* Refactor UI
* Introduce render_video_preview()
* Add preview back part1
* Add preview back part2, Introduce --video-quality for CLI
* Get the preview working
* Couple if minor UI fixes
* Add video encoder via CLI
* Change default video quality, Integrate recent directories for UI
* Move temporary files to temp/{target-name}
* Fix fps detection
* Rename method
* Introduce suggest methods for args defaults, output mode and core/threads count via postfix
* Fix max_memory and output memory in progress bar too
* Turns out mac has a different memory unit
* Add typing to swapper
* Fix FileNotFoundError while deleting temp
* Updated requirements.txt for macs.
(cherry picked from commit fd00a18)
* Doing bunch of renaming and typing
* Just a cosmetic
* Doing bunch of renaming again
* Introduce execution provider to support DirectML
* enhancer update
* remove useless code
* remove useless code
* remove useless code
* fix
* reslove some errors in code review.
* methods rename
* del Unused import
* recover the conditional installation for darwin!
* recover the import module
* del try catch in unrelate codes
* fix error in argument and potential infinity loop
* remove the ROCM check before face-enhancer
* fix lint error
* add the process for image
* conditional process according to --frame-processor
* Hotfix usage of --frame-processor face-swapper face-enhancer
* Auto download models
* typo fixed
* Fix framerate and audio sync issues
* Limit the video encoders according to -crf support
* Limit the video encoders according to -crf support part2
* Migrate to theme based UI using customtkinter
* Show full preview according to video frames total
* Simplify is_image and is_video, close preview on source/target change, show preview slider on video only, fix start button error
* Fix linter
* Use float over int for more accurate fps
* introduce a pre_check() to enhancer...
* update
* Update utilities.py
* move the model_path to the method
* Fix model paths
* Fix linter
* Fix images scaling
* Update onnxruntime-silicon and monkey patch ssl for mac
* Downgrade onnxruntime-silicon again
* Introduce a first version of CONTRIBUTING.md
* Update CONTRIBUTING.md
* Add libvpx-vp9 to video encoder's choices
* Update CONTRIBUTING.md
* Migrate gpu flags to execution flags
* Fix linter
* Encode and decode execution providers for easier usage
* Fix comment
* Update CLI usage
* Fixed wrong frame colors for preview
* Introduce dynamic frame procesors
* Remove unused imports
* Use different structure
* modified: roop/core.py , enhancer and swapper
* fix link error
* modified core.py ui.py frame/core.py
* fix get_frame_processors_modules()
* fix lint error in core.py
* fix face_enhancer.py
* fix enhancer.py
* fix ui.py
* modified: roop/ui.py
* fix ui.py
* Update ui.py
* Fix preview to work with multiple frame processors
* Remove multi processing as the performance is equal but memory consumtion better
* Extract common methods to processors.frame.core
* Add comments and todos
* Minor fixes
* Minor fixes
* Limit enhancer to 1 thread but restore afterwards
* Limit enhancer to 1 thread but restore afterwards
* Update README and GUI demo
* Implementation check for frame processors
* Improve path validation
* Introduce normalize output path
* Fix GUI startup
* Introduce pre_start() and move globals check to pre_check()
* Flip the hooks and update CLI usage
* Introduce bool returns for pre_check() and pre_start(), Scope for terminal output
* Cast deprecated args and warn the user
* Migrate to ctk.CTkImage
* readme update: old keys fixed
* Unused import made CI test to fail
* Give gfpgan a shot
* Update dependencies
* Update dependencies
* Update dependencies
* Update dependencies
* Fix ci (#498)
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Fix preview (#499)
* Minor changes
* Fix override of files using restore_audio()
* Host the models in our huggingface space
* Remove everything source face related from enhancer (#510)
* Remove everything source face related from enhancer
* Disable upscale for enhancer to double fps
* Using futures for multi threading
* Introduce predicter (#530)
* Hotfix predicter
* Fix square brackets in the target path (#532)
* fixes the problem with square brackets in the target file path
* fixes the problem with square brackets in the target file path
* Ok, here is the fix without regexps
* unused import
* fix for ci
* glob.escape fits here
* Make multiple processors work with images
* Fix output normalize for deprecated args
* Make roop more or less type-safe (#541)
* Make roop more or less type-safe
* Fix ci.yml
* Fix urllib type error
* Rename globals in ui
* Update utilities.py (#542)
Updated the extraction process with the ffmpeg command that corrects the colorspace while extracting to png, and corrected the ffmpeg command, adding '-pix_fmt', 'rgb24', '-sws_flags', '+accurate_rnd+full_chroma_int', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1'
'-pix_fmt rgb24', means treat the image as RGB (or RGBA)
'-sws_flags +accurate_rnd+full_chroma_int', means use full color and chroma subsampling instead of 4:2:0
'-colorspace 1', '-color_primaries 1', '-color_trc 1' put the metadata color tags to the png
* Use GFPGANv1.4 for enhancer
* Fixing the colorspace issue when writing the mp4 from the extracted pngs (#550)
* Update utilities.py
Updated the extraction process with the ffmpeg command that corrects the colorspace while extracting to png, and corrected the ffmpeg command, adding '-pix_fmt', 'rgb24', '-sws_flags', '+accurate_rnd+full_chroma_int', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1'
'-pix_fmt rgb24', means treat the image as RGB (or RGBA)
'-sws_flags +accurate_rnd+full_chroma_int', means use full color and chroma subsampling instead of 4:2:0
'-colorspace 1', '-color_primaries 1', '-color_trc 1' put the metadata color tags to the png
* Fixing color conversion from temp png sequence to mp4
'-sws_flags', 'spline+accurate_rnd+full_chroma_int', ' use full color and chroma subsampling
-vf', 'colorspace=bt709:iall=bt601-6-625:fast=1', keep the same rec709 colorspace
'-color_range', '1', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1', put the metadata color tags to the mp4
* Revert "Fixing the colorspace issue when writing the mp4 from the extracted pngs (#550)"
This reverts commit cf5f27d.
* Revert "Update utilities.py (#542)"
This reverts commit d57279c.
* Restore colorspace restoring
* Add metadata to cli and ui
* Introduce Face and Frame typing
* Update CLI usage
---------
Co-authored-by: Phan Tuấn Anh <[email protected]>
Co-authored-by: Antoine Buchser <[email protected]>
Co-authored-by: Eamonn A. Sweeney <[email protected]>
Co-authored-by: Moeblack <Moeblack@[email protected]>
Co-authored-by: Pozitronik <[email protected]>
Co-authored-by: Pikachu~~~ <[email protected]>
Co-authored-by: K1llM@n <[email protected]>
Co-authored-by: NickPittas <[email protected]>
* ffmpeg platform-agnostic hardware-acceleration
* clear CUDA cache after swapping
on low VRAM + ffmpeg cuda acceleration, clearing cache prevent cuda out-of-memory error
* check torch gpu before clearing cache
* torch check nvidia only
* syntax error
* Adjust comment
* Normalize ARGS
* Remove path normalization
* Remove args overrides
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Revert to Ubuntu test only as Windows hangs
* Simplified the way to maintain aspect ratio of the preview, and maintaining aspect ratio of the miniatures
* Change face and target images from contain to fit
* Improve status output
* Massive utilities and core refactoring
* Fix sound
* Fix sound part2
* Fix more
* Move every UI related thing to ui.py
* Refactor UI
* Introduce render_video_preview()
* Add preview back part1
* Add preview back part2, Introduce --video-quality for CLI
* Get the preview working
* Couple if minor UI fixes
* Add video encoder via CLI
* Change default video quality, Integrate recent directories for UI
* Move temporary files to temp/{target-name}
* Fix fps detection
* Rename method
* Introduce suggest methods for args defaults, output mode and core/threads count via postfix
* Fix max_memory and output memory in progress bar too
* Turns out mac has a different memory unit
* Add typing to swapper
* Fix FileNotFoundError while deleting temp
* Updated requirements.txt for macs.
(cherry picked from commit 47b55e8)
* Doing bunch of renaming and typing
* Just a cosmetic
* Doing bunch of renaming again
* Introduce execution provider to support DirectML
* enhancer update
* remove useless code
* remove useless code
* remove useless code
* fix
* reslove some errors in code review.
* methods rename
* del Unused import
* recover the conditional installation for darwin!
* recover the import module
* del try catch in unrelate codes
* fix error in argument and potential infinity loop
* remove the ROCM check before face-enhancer
* fix lint error
* add the process for image
* conditional process according to --frame-processor
* Hotfix usage of --frame-processor face-swapper face-enhancer
* Auto download models
* typo fixed
* Fix framerate and audio sync issues
* Limit the video encoders according to -crf support
* Limit the video encoders according to -crf support part2
* Migrate to theme based UI using customtkinter
* Show full preview according to video frames total
* Simplify is_image and is_video, close preview on source/target change, show preview slider on video only, fix start button error
* Fix linter
* Use float over int for more accurate fps
* introduce a pre_check() to enhancer...
* update
* Update utilities.py
* move the model_path to the method
* Fix model paths
* Fix linter
* Fix images scaling
* Update onnxruntime-silicon and monkey patch ssl for mac
* Downgrade onnxruntime-silicon again
* Introduce a first version of CONTRIBUTING.md
* Update CONTRIBUTING.md
* Add libvpx-vp9 to video encoder's choices
* Update CONTRIBUTING.md
* Migrate gpu flags to execution flags
* Fix linter
* Encode and decode execution providers for easier usage
* Fix comment
* Update CLI usage
* Fixed wrong frame colors for preview
* Introduce dynamic frame procesors
* Remove unused imports
* Use different structure
* modified: roop/core.py , enhancer and swapper
* fix link error
* modified core.py ui.py frame/core.py
* fix get_frame_processors_modules()
* fix lint error in core.py
* fix face_enhancer.py
* fix enhancer.py
* fix ui.py
* modified: roop/ui.py
* fix ui.py
* Update ui.py
* Fix preview to work with multiple frame processors
* Remove multi processing as the performance is equal but memory consumtion better
* Extract common methods to processors.frame.core
* Add comments and todos
* Minor fixes
* Minor fixes
* Limit enhancer to 1 thread but restore afterwards
* Limit enhancer to 1 thread but restore afterwards
* Update README and GUI demo
* Implementation check for frame processors
* Improve path validation
* Introduce normalize output path
* Fix GUI startup
* Introduce pre_start() and move globals check to pre_check()
* Flip the hooks and update CLI usage
* Introduce bool returns for pre_check() and pre_start(), Scope for terminal output
* Cast deprecated args and warn the user
* Migrate to ctk.CTkImage
* readme update: old keys fixed
* Unused import made CI test to fail
* Give gfpgan a shot
* Update dependencies
* Update dependencies
* Update dependencies
* Update dependencies
* Fix ci (s0md3v#498)
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Fix preview (s0md3v#499)
* Minor changes
* Fix override of files using restore_audio()
* Host the models in our huggingface space
* Remove everything source face related from enhancer (s0md3v#510)
* Remove everything source face related from enhancer
* Disable upscale for enhancer to double fps
* Using futures for multi threading
* Introduce predicter (s0md3v#530)
* Hotfix predicter
* Fix square brackets in the target path (s0md3v#532)
* fixes the problem with square brackets in the target file path
* fixes the problem with square brackets in the target file path
* Ok, here is the fix without regexps
* unused import
* fix for ci
* glob.escape fits here
* Make multiple processors work with images
* Fix output normalize for deprecated args
* Make roop more or less type-safe (s0md3v#541)
* Make roop more or less type-safe
* Fix ci.yml
* Fix urllib type error
* Rename globals in ui
* Update utilities.py (s0md3v#542)
Updated the extraction process with the ffmpeg command that corrects the colorspace while extracting to png, and corrected the ffmpeg command, adding '-pix_fmt', 'rgb24', '-sws_flags', '+accurate_rnd+full_chroma_int', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1'
'-pix_fmt rgb24', means treat the image as RGB (or RGBA)
'-sws_flags +accurate_rnd+full_chroma_int', means use full color and chroma subsampling instead of 4:2:0
'-colorspace 1', '-color_primaries 1', '-color_trc 1' put the metadata color tags to the png
* Use GFPGANv1.4 for enhancer
* Fixing the colorspace issue when writing the mp4 from the extracted pngs (s0md3v#550)
* Update utilities.py
Updated the extraction process with the ffmpeg command that corrects the colorspace while extracting to png, and corrected the ffmpeg command, adding '-pix_fmt', 'rgb24', '-sws_flags', '+accurate_rnd+full_chroma_int', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1'
'-pix_fmt rgb24', means treat the image as RGB (or RGBA)
'-sws_flags +accurate_rnd+full_chroma_int', means use full color and chroma subsampling instead of 4:2:0
'-colorspace 1', '-color_primaries 1', '-color_trc 1' put the metadata color tags to the png
* Fixing color conversion from temp png sequence to mp4
'-sws_flags', 'spline+accurate_rnd+full_chroma_int', ' use full color and chroma subsampling
-vf', 'colorspace=bt709:iall=bt601-6-625:fast=1', keep the same rec709 colorspace
'-color_range', '1', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1', put the metadata color tags to the mp4
* Revert "Fixing the colorspace issue when writing the mp4 from the extracted pngs (s0md3v#550)"
This reverts commit cf5f27d.
* Revert "Update utilities.py (s0md3v#542)"
This reverts commit d57279c.
* Restore colorspace restoring
* Add metadata to cli and ui
* Introduce Face and Frame typing
* Update CLI usage
---------
Co-authored-by: Phan Tuấn Anh <[email protected]>
Co-authored-by: Antoine Buchser <[email protected]>
Co-authored-by: Eamonn A. Sweeney <[email protected]>
Co-authored-by: Moeblack <Moeblack@[email protected]>
Co-authored-by: Pozitronik <[email protected]>
Co-authored-by: Pikachu~~~ <[email protected]>
Co-authored-by: K1llM@n <[email protected]>
Co-authored-by: NickPittas <[email protected]>
* ffmpeg platform-agnostic hardware-acceleration
* clear CUDA cache after swapping
on low VRAM + ffmpeg cuda acceleration, clearing cache prevent cuda out-of-memory error
* check torch gpu before clearing cache
* torch check nvidia only
* syntax error
* Adjust comment
* Normalize ARGS
* Remove path normalization
* Remove args overrides
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Revert to Ubuntu test only as Windows hangs
* Simplified the way to maintain aspect ratio of the preview, and maintaining aspect ratio of the miniatures
* Change face and target images from contain to fit
* Improve status output
* Massive utilities and core refactoring
* Fix sound
* Fix sound part2
* Fix more
* Move every UI related thing to ui.py
* Refactor UI
* Introduce render_video_preview()
* Add preview back part1
* Add preview back part2, Introduce --video-quality for CLI
* Get the preview working
* Couple if minor UI fixes
* Add video encoder via CLI
* Change default video quality, Integrate recent directories for UI
* Move temporary files to temp/{target-name}
* Fix fps detection
* Rename method
* Introduce suggest methods for args defaults, output mode and core/threads count via postfix
* Fix max_memory and output memory in progress bar too
* Turns out mac has a different memory unit
* Add typing to swapper
* Fix FileNotFoundError while deleting temp
* Updated requirements.txt for macs.
(cherry picked from commit 47b55e8)
* Doing bunch of renaming and typing
* Just a cosmetic
* Doing bunch of renaming again
* Introduce execution provider to support DirectML
* enhancer update
* remove useless code
* remove useless code
* remove useless code
* fix
* reslove some errors in code review.
* methods rename
* del Unused import
* recover the conditional installation for darwin!
* recover the import module
* del try catch in unrelate codes
* fix error in argument and potential infinity loop
* remove the ROCM check before face-enhancer
* fix lint error
* add the process for image
* conditional process according to --frame-processor
* Hotfix usage of --frame-processor face-swapper face-enhancer
* Auto download models
* typo fixed
* Fix framerate and audio sync issues
* Limit the video encoders according to -crf support
* Limit the video encoders according to -crf support part2
* Migrate to theme based UI using customtkinter
* Show full preview according to video frames total
* Simplify is_image and is_video, close preview on source/target change, show preview slider on video only, fix start button error
* Fix linter
* Use float over int for more accurate fps
* introduce a pre_check() to enhancer...
* update
* Update utilities.py
* move the model_path to the method
* Fix model paths
* Fix linter
* Fix images scaling
* Update onnxruntime-silicon and monkey patch ssl for mac
* Downgrade onnxruntime-silicon again
* Introduce a first version of CONTRIBUTING.md
* Update CONTRIBUTING.md
* Add libvpx-vp9 to video encoder's choices
* Update CONTRIBUTING.md
* Migrate gpu flags to execution flags
* Fix linter
* Encode and decode execution providers for easier usage
* Fix comment
* Update CLI usage
* Fixed wrong frame colors for preview
* Introduce dynamic frame procesors
* Remove unused imports
* Use different structure
* modified: roop/core.py , enhancer and swapper
* fix link error
* modified core.py ui.py frame/core.py
* fix get_frame_processors_modules()
* fix lint error in core.py
* fix face_enhancer.py
* fix enhancer.py
* fix ui.py
* modified: roop/ui.py
* fix ui.py
* Update ui.py
* Fix preview to work with multiple frame processors
* Remove multi processing as the performance is equal but memory consumtion better
* Extract common methods to processors.frame.core
* Add comments and todos
* Minor fixes
* Minor fixes
* Limit enhancer to 1 thread but restore afterwards
* Limit enhancer to 1 thread but restore afterwards
* Update README and GUI demo
* Implementation check for frame processors
* Improve path validation
* Introduce normalize output path
* Fix GUI startup
* Introduce pre_start() and move globals check to pre_check()
* Flip the hooks and update CLI usage
* Introduce bool returns for pre_check() and pre_start(), Scope for terminal output
* Cast deprecated args and warn the user
* Migrate to ctk.CTkImage
* readme update: old keys fixed
* Unused import made CI test to fail
* Give gfpgan a shot
* Update dependencies
* Update dependencies
* Update dependencies
* Update dependencies
* Fix ci (s0md3v#498)
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Fix preview (s0md3v#499)
* Minor changes
* Fix override of files using restore_audio()
* Host the models in our huggingface space
* Remove everything source face related from enhancer (s0md3v#510)
* Remove everything source face related from enhancer
* Disable upscale for enhancer to double fps
* Using futures for multi threading
* Introduce predicter (s0md3v#530)
* Hotfix predicter
* Fix square brackets in the target path (s0md3v#532)
* fixes the problem with square brackets in the target file path
* fixes the problem with square brackets in the target file path
* Ok, here is the fix without regexps
* unused import
* fix for ci
* glob.escape fits here
* Make multiple processors work with images
* Fix output normalize for deprecated args
* Make roop more or less type-safe (s0md3v#541)
* Make roop more or less type-safe
* Fix ci.yml
* Fix urllib type error
* Rename globals in ui
* Update utilities.py (s0md3v#542)
Updated the extraction process with the ffmpeg command that corrects the colorspace while extracting to png, and corrected the ffmpeg command, adding '-pix_fmt', 'rgb24', '-sws_flags', '+accurate_rnd+full_chroma_int', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1'
'-pix_fmt rgb24', means treat the image as RGB (or RGBA)
'-sws_flags +accurate_rnd+full_chroma_int', means use full color and chroma subsampling instead of 4:2:0
'-colorspace 1', '-color_primaries 1', '-color_trc 1' put the metadata color tags to the png
* Use GFPGANv1.4 for enhancer
* Fixing the colorspace issue when writing the mp4 from the extracted pngs (s0md3v#550)
* Update utilities.py
Updated the extraction process with the ffmpeg command that corrects the colorspace while extracting to png, and corrected the ffmpeg command, adding '-pix_fmt', 'rgb24', '-sws_flags', '+accurate_rnd+full_chroma_int', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1'
'-pix_fmt rgb24', means treat the image as RGB (or RGBA)
'-sws_flags +accurate_rnd+full_chroma_int', means use full color and chroma subsampling instead of 4:2:0
'-colorspace 1', '-color_primaries 1', '-color_trc 1' put the metadata color tags to the png
* Fixing color conversion from temp png sequence to mp4
'-sws_flags', 'spline+accurate_rnd+full_chroma_int', ' use full color and chroma subsampling
-vf', 'colorspace=bt709:iall=bt601-6-625:fast=1', keep the same rec709 colorspace
'-color_range', '1', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1', put the metadata color tags to the mp4
* Revert "Fixing the colorspace issue when writing the mp4 from the extracted pngs (s0md3v#550)"
This reverts commit cf5f27d.
* Revert "Update utilities.py (s0md3v#542)"
This reverts commit d57279c.
* Restore colorspace restoring
* Add metadata to cli and ui
* Introduce Face and Frame typing
* Update CLI usage
---------
Co-authored-by: Phan Tuấn Anh <[email protected]>
Co-authored-by: Antoine Buchser <[email protected]>
Co-authored-by: Eamonn A. Sweeney <[email protected]>
Co-authored-by: Moeblack <Moeblack@[email protected]>
Co-authored-by: Pozitronik <[email protected]>
Co-authored-by: Pikachu~~~ <[email protected]>
Co-authored-by: K1llM@n <[email protected]>
Co-authored-by: NickPittas <[email protected]>
* ffmpeg platform-agnostic hardware-acceleration
* clear CUDA cache after swapping
on low VRAM + ffmpeg cuda acceleration, clearing cache prevent cuda out-of-memory error
* check torch gpu before clearing cache
* torch check nvidia only
* syntax error
* Adjust comment
* Normalize ARGS
* Remove path normalization
* Remove args overrides
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Run test on Linux and Windows
* Revert to Ubuntu test only as Windows hangs
* Simplified the way to maintain aspect ratio of the preview, and maintaining aspect ratio of the miniatures
* Change face and target images from contain to fit
* Improve status output
* Massive utilities and core refactoring
* Fix sound
* Fix sound part2
* Fix more
* Move every UI related thing to ui.py
* Refactor UI
* Introduce render_video_preview()
* Add preview back part1
* Add preview back part2, Introduce --video-quality for CLI
* Get the preview working
* Couple if minor UI fixes
* Add video encoder via CLI
* Change default video quality, Integrate recent directories for UI
* Move temporary files to temp/{target-name}
* Fix fps detection
* Rename method
* Introduce suggest methods for args defaults, output mode and core/threads count via postfix
* Fix max_memory and output memory in progress bar too
* Turns out mac has a different memory unit
* Add typing to swapper
* Fix FileNotFoundError while deleting temp
* Updated requirements.txt for macs.
(cherry picked from commit c63cf0c)
* Doing bunch of renaming and typing
* Just a cosmetic
* Doing bunch of renaming again
* Introduce execution provider to support DirectML
* enhancer update
* remove useless code
* remove useless code
* remove useless code
* fix
* reslove some errors in code review.
* methods rename
* del Unused import
* recover the conditional installation for darwin!
* recover the import module
* del try catch in unrelate codes
* fix error in argument and potential infinity loop
* remove the ROCM check before face-enhancer
* fix lint error
* add the process for image
* conditional process according to --frame-processor
* Hotfix usage of --frame-processor face-swapper face-enhancer
* Auto download models
* typo fixed
* Fix framerate and audio sync issues
* Limit the video encoders according to -crf support
* Limit the video encoders according to -crf support part2
* Migrate to theme based UI using customtkinter
* Show full preview according to video frames total
* Simplify is_image and is_video, close preview on source/target change, show preview slider on video only, fix start button error
* Fix linter
* Use float over int for more accurate fps
* introduce a pre_check() to enhancer...
* update
* Update utilities.py
* move the model_path to the method
* Fix model paths
* Fix linter
* Fix images scaling
* Update onnxruntime-silicon and monkey patch ssl for mac
* Downgrade onnxruntime-silicon again
* Introduce a first version of CONTRIBUTING.md
* Update CONTRIBUTING.md
* Add libvpx-vp9 to video encoder's choices
* Update CONTRIBUTING.md
* Migrate gpu flags to execution flags
* Fix linter
* Encode and decode execution providers for easier usage
* Fix comment
* Update CLI usage
* Fixed wrong frame colors for preview
* Introduce dynamic frame procesors
* Remove unused imports
* Use different structure
* modified: roop/core.py , enhancer and swapper
* fix link error
* modified core.py ui.py frame/core.py
* fix get_frame_processors_modules()
* fix lint error in core.py
* fix face_enhancer.py
* fix enhancer.py
* fix ui.py
* modified: roop/ui.py
* fix ui.py
* Update ui.py
* Fix preview to work with multiple frame processors
* Remove multi processing as the performance is equal but memory consumtion better
* Extract common methods to processors.frame.core
* Add comments and todos
* Minor fixes
* Minor fixes
* Limit enhancer to 1 thread but restore afterwards
* Limit enhancer to 1 thread but restore afterwards
* Update README and GUI demo
* Implementation check for frame processors
* Improve path validation
* Introduce normalize output path
* Fix GUI startup
* Introduce pre_start() and move globals check to pre_check()
* Flip the hooks and update CLI usage
* Introduce bool returns for pre_check() and pre_start(), Scope for terminal output
* Cast deprecated args and warn the user
* Migrate to ctk.CTkImage
* readme update: old keys fixed
* Unused import made CI test to fail
* Give gfpgan a shot
* Update dependencies
* Update dependencies
* Update dependencies
* Update dependencies
* Fix ci (s0md3v#498)
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Use different dependencies for CI
* Fix preview (s0md3v#499)
* Minor changes
* Fix override of files using restore_audio()
* Host the models in our huggingface space
* Remove everything source face related from enhancer (s0md3v#510)
* Remove everything source face related from enhancer
* Disable upscale for enhancer to double fps
* Using futures for multi threading
* Introduce predicter (s0md3v#530)
* Hotfix predicter
* Fix square brackets in the target path (s0md3v#532)
* fixes the problem with square brackets in the target file path
* fixes the problem with square brackets in the target file path
* Ok, here is the fix without regexps
* unused import
* fix for ci
* glob.escape fits here
* Make multiple processors work with images
* Fix output normalize for deprecated args
* Make roop more or less type-safe (s0md3v#541)
* Make roop more or less type-safe
* Fix ci.yml
* Fix urllib type error
* Rename globals in ui
* Update utilities.py (s0md3v#542)
Updated the extraction process with the ffmpeg command that corrects the colorspace while extracting to png, and corrected the ffmpeg command, adding '-pix_fmt', 'rgb24', '-sws_flags', '+accurate_rnd+full_chroma_int', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1'
'-pix_fmt rgb24', means treat the image as RGB (or RGBA)
'-sws_flags +accurate_rnd+full_chroma_int', means use full color and chroma subsampling instead of 4:2:0
'-colorspace 1', '-color_primaries 1', '-color_trc 1' put the metadata color tags to the png
* Use GFPGANv1.4 for enhancer
* Fixing the colorspace issue when writing the mp4 from the extracted pngs (s0md3v#550)
* Update utilities.py
Updated the extraction process with the ffmpeg command that corrects the colorspace while extracting to png, and corrected the ffmpeg command, adding '-pix_fmt', 'rgb24', '-sws_flags', '+accurate_rnd+full_chroma_int', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1'
'-pix_fmt rgb24', means treat the image as RGB (or RGBA)
'-sws_flags +accurate_rnd+full_chroma_int', means use full color and chroma subsampling instead of 4:2:0
'-colorspace 1', '-color_primaries 1', '-color_trc 1' put the metadata color tags to the png
* Fixing color conversion from temp png sequence to mp4
'-sws_flags', 'spline+accurate_rnd+full_chroma_int', ' use full color and chroma subsampling
-vf', 'colorspace=bt709:iall=bt601-6-625:fast=1', keep the same rec709 colorspace
'-color_range', '1', '-colorspace', '1', '-color_primaries', '1', '-color_trc', '1', put the metadata color tags to the mp4
* Revert "Fixing the colorspace issue when writing the mp4 from the extracted pngs (s0md3v#550)"
This reverts commit cf5f27d.
* Revert "Update utilities.py (s0md3v#542)"
This reverts commit d57279c.
* Restore colorspace restoring
* Add metadata to cli and ui
* Introduce Face and Frame typing
* Update CLI usage
---------
Co-authored-by: Phan Tuấn Anh <[email protected]>
Co-authored-by: Antoine Buchser <[email protected]>
Co-authored-by: Eamonn A. Sweeney <[email protected]>
Co-authored-by: Moeblack <Moeblack@[email protected]>
Co-authored-by: Pozitronik <[email protected]>
Co-authored-by: Pikachu~~~ <[email protected]>
Co-authored-by: K1llM@n <[email protected]>
Co-authored-by: NickPittas <[email protected]>
When the target path contains square brackets symbols, swapping process fails without any warning messages.
It is happened because the
glob.glob()method interprets square brackets as a character class groups.This PR fixes it.