Commit Graph

705 Commits

Author SHA1 Message Date
henryruhs 8a9c596fde merge master into v4 - post adjustments 2026-07-01 13:34:54 +02:00
Henry Ruhs 300470e7c7 port/master-into-v4 (#1176)
* 3.7.0 (#1175)

* mark as next, introduce dynamic scale for face debugger

* use latest onnxruntime

* update within Gradio 5

* Remove system memory limit (#986)

* remove system memory limit from ui

* remove system memory limit from args.py

* flatten the face store

* prevent countless importlib.import_module calls

* remove --onnxruntime from install.py

* remove --onnxruntime from install.py

* resolve static inference providers to fix macos (#1127)

* resolve static inference providers to fix macos

* fix lint

* restore old behaviour

* restore old behaviour

* handle ghost and uniface as well

* adjust condition for ghost and uniface

* fix Gradio gallery styles

* remove face store (#1132)

* fix dataflow in streamer

* Face selector auto mode (#1137)

* introduce face selector auto mode

* introduce face selector auto mode

* introduce face selector auto mode

* correct way is to pass source_vision_frames

* make the world a better place

* fix dataflow in faceswapper, no read of files withing inner methods (#1148)

* fix dataflow in faceswapper, no read of files withing inner methods

* fix lint

* adjust code more

* adjust code more

* bring back the face store but for source and reference only (#1149)

* bring back the face store but for source and reference only

* fix ci

* minor improvement

* guard for tobytes()

* drop condition in select_faces()

* Replace CONFIG_PARSER global with @lru_cache (#1147)

* remove global config_parser

* fix import order

* remove lambda

* remove unused block

* optimize app context detection

* decouple common modules from core (#1152)

* decouple common modules from core

* remove that nonsense

* remove that nonsense

* minor adjustment to workflows

* Tag HEVC output as hvc1 and move moov atom to the front (#1153)

* Tag HEVC output as hvc1 and move moov atom to the front

ffmpeg defaults HEVC in MP4 to the 'hev1' sample entry and leaves the moov
atom at the tail. Apple players (QuickTime, Finder QuickLook) refuse to decode
'hev1' and stall reading a tail-placed moov on large files, so hevc_nvenc /
libx265 renders cannot be previewed on macOS.

- add ffmpeg_builder.set_video_tag(): emit `-tag:v hvc1` for every HEVC
  encoder (libx265, hevc_nvenc, hevc_amf, hevc_qsv, hevc_videotoolbox).
  Applied in merge_video where the encoder is known; `-c:v copy` in the audio
  mux / concat steps preserves the tag.
- add ffmpeg_builder.set_faststart(): emit `-movflags +faststart`, applied in
  restore_audio / replace_audio / concat_video which write the final output.

H.264 and other codecs are left untouched. Verified on a real hevc_nvenc
render: hev1 hung QuickLook (no thumbnail); after the patch the file is hvc1
with a front-placed moov and QuickLook generates a thumbnail.

* Restrict hvc1 tag and faststart to quicktime containers

Gate set_video_tag / set_faststart on the output container format
(m4v, mov, mp4) via get_file_format(), so non-quicktime muxers no longer
receive -tag:v hvc1 / -movflags +faststart. Trim test_set_video_tag to a
single positive and negative assertion.

Addresses review on #1153.

* Move hvc1 tag and faststart gates into ffmpeg_builder

Rename set_video_tag / set_faststart to conditional_* and push the
container-format gate (m4v, mov, mp4) inside the builders, keeping
ffmpeg.py free of inline conditionals. Matches the set_image_quality
pattern. Addresses review on #1153.

* post cleanup after merge

* Pack target frames (#1158)

* pack target frames

* add todos

* add todos, resolve todos

* resolve todos

* change names

* revert to single target frame for select faces

* fix lint

* return empty frame

* get() have no default

* Fix trim (#1162)

* fix trim

* fix trim

* rename ffmpeg builder method

* rename to temp_frame_set and temp_frame_pattern

---------

Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com>
Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com>

* Implement face tracker (#1163)

* add face tracker

* change get_nearest_track_face -> get_nearest_track_index

* create face_creator.py and move methods around

* add type FaceTrack

* naming

* remove iou test, don't belong there

* fix spaces

* rename to interpolate_points

* rename to find_best_face_track

* just track_faces

* cleanp

* previous next naming

* remove >= and >=

* rename

* remove helper from test and use face from source.jpg

* make get_anchor_indices more readable

* track_faces() call before and is forwarded to select_faces

* change to interpolate_faces

* rename methods

* rename methods

* rename variables

* remove dtype

* move face_anlyser -> face_creator

* claenup face_creator.py

* move tests to dedicated test face detector

* move tracking inside select_faces

* simplify face_tracker (#1165)

* minor renaming

* improve face_tracker test (#1166)

* improve face_tracker test

* cleanup

* Add target frame amount (#1167)

* introduce --target-frame-amount

* add ui

* make track_faces conditional

* update choices.py

* fix []

* rename component file to frame_process.py

* fix track preview (#1168)

* introduce face origin (#1169)

* add guard to prevent failure

* show and hide voice extractor according to lip syncer

* rename average_face_coordinates to average_face_geometry

* use static faces for select_faces()

* face store with lock (#1171)

* face store with lock

* face store with lock

* remove refill color from bbox

* adjust tests and handle frame_position proper way

* enforce similar naming

* introduce face tracker score

* introduce face tracker score

* fix/audio-trim-alignment (#1173)

* fix audio offset

* fix audio offset

* remove reference_frame_number check

---------

Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com>

* reduce face tracker score from 0 to 0.5

* mark as 3.7.0

* make face tracker stateless

---------

Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com>
Co-authored-by: kazuki nakai <kazuki.nakai@agiletec.net>
Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com>

* update preview

* fix wording

* fix wording

* last minute change to frame distribution

---------

Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com>
Co-authored-by: kazuki nakai <kazuki.nakai@agiletec.net>
Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com>
2026-07-01 11:51:54 +02:00
Henry Ruhs 5f403ada65 remove opus frame size (#1160)
* remove opus frame size

* improve naming, follow audio.py
2026-06-15 19:19:51 +02:00
henryruhs c6aed91698 use dynamic audio frame size 2026-06-15 17:14:24 +02:00
Henry Ruhs 2854bf1db0 Resolve final stream TODOs (#1155)
* refactor a lot

* refactor a lot

* fix test

* fix test
2026-06-15 11:37:25 +02:00
Henry Ruhs d5271d21a1 minor adjustment for workflow setup and clear (#1157)
* minor adjustment for workflow setup and clear

* minor adjustment for workflow setup and clear
2026-06-14 22:57:01 +02:00
Henry Ruhs e7d22e84bf use Buffer and BufferPack everywhere (#1154) 2026-06-11 13:03:29 +02:00
henryruhs 57189c638e optimize app context detection 2026-06-09 22:30:37 +02:00
Henry Ruhs 00fb89d4f1 Best performance to code ratio for stream (#1150)
* queue with futures, kill deque, add couple of todos

* resolve couple of todos

* add more todos

* add more todos and resolve others

* add more todos and resolve others

* fix test

* fix collapse

* adjust naming a bit
2026-06-09 22:16:51 +02:00
henryruhs 67cc3de934 just ignore type 2026-06-06 15:31:14 +02:00
henryruhs accad0a2cb fix ci 2026-06-06 13:53:30 +02:00
henryruhs 46d1575c9b fix ci 2026-06-06 13:41:28 +02:00
henryruhs f3bc5ffb4b changes for vp9 support 2026-06-06 12:39:14 +02:00
henryruhs 5287ce6dcb add todos 2026-06-06 11:45:38 +02:00
henryruhs 5d6258e17e add vp9 support 2026-06-06 11:39:53 +02:00
henryruhs 036c5c0225 add vp9 support 2026-06-06 11:15:00 +02:00
henryruhs 2e884941f8 resolve todos 2026-06-06 08:36:20 +02:00
henryruhs 0ef5de1d02 guard for tobytes()
(cherry picked from commit 0cbfa5a415)
2026-06-05 21:47:57 +02:00
henryruhs dcb9a7a59b minor improvement
(cherry picked from commit fd3e8b82e2)
2026-06-05 21:47:57 +02:00
henryruhs 3f6738dd1d fix ci
(cherry picked from commit fbfb21df3b)
2026-06-05 21:47:57 +02:00
henryruhs 19f8f5f206 bring back the face store but for source and reference only
(cherry picked from commit 9d8a5be714)
2026-06-05 21:47:57 +02:00
Henry Ruhs 775985645e Push based receive with queue (#1146)
* move to push based receive

* move to push based receive, fix mocks

* fix tests

* add todos

* remove asyncio

* remove asyncio

* resolve todos

* move to queue without events

* prevent debug spam

* concurrent stream inference

stream_video.py: pipeline face-swap inference across execution_thread_count workers (ThreadPoolExecutor + bounded in-flight deque, ordered encode) to keep the GPU busy during encode

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

* add todos

* add todos

* add missing state

---------

Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-05 00:18:57 +02:00
henryruhs 1f494f54db fix hashes 2026-06-03 10:43:16 +02:00
henryruhs 476a21cc7a fix hashes 2026-06-03 10:43:09 +02:00
henryruhs 121c3a5af6 update hashes 2026-06-03 10:36:00 +02:00
Henry Ruhs 2ac9b70550 Feat/remb both direction (#1145)
* revamp remb to take both directions

* extract more methods
2026-06-03 10:31:33 +02:00
henryruhs d9553b12e8 remove copy of buffer 2026-06-02 17:10:25 +02:00
henryruhs 3bda73699c fix fast seeking 2026-06-02 16:29:58 +02:00
Henry Ruhs 7181b41f2d Feat/finalize stream (#1144)
* break stream helper into pieces

* remove todos
2026-06-02 16:05:21 +02:00
harisreedhar 314ee61826 add create_event method 2026-06-02 03:06:16 +05:30
Harisreedhar beeb1d99e9 Refactor(stream-helper): split encode/receive loops and unify audio/video structure (#1142)
* split encode loop

* unify audio and video encode loop methods

* improve variable names

* split receive methods

* test improve

* try to avoid != with even more weird approach

* remove source_path check

* remove empty variables

* fix lint

* avoid not in condition

* rename

* fix lint
2026-06-02 01:50:40 +05:30
Henry Ruhs e2e4e6a95b Refactor/stream helper testing (#1141)
* fix tests part1

* improve testing

* use real audio

* more renaming

* switch to hash assertions

* switch to hash assertions

* switch to hash assertions

* switch to hash assertions

* skip on macos
2026-06-01 18:42:37 +02:00
Harisreedhar 3eeb505c86 Refactor(stream-helper): pass peer context objects to receivers, fix codec guard, move YUV conversion into decoders (#1140)
* extract numpy.empty(0) into an empty_vision_frame variable so the sentinel intent is clear

* bring back todos

* bring back todos

* rename opus_buffer to audio

* move reshape and cvtColor into decoder modules

* add audio_codec check

* add audio_codec check

* rename audio video buffer

* simplify receive methods and cleanup test

* todo

* revert decode methods to return pointer

* remove duplicate test_stream_helper — tests live in test_api_stream_helper

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

---------

Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-06-01 17:21:06 +05:30
harisreedhar 02f684e220 bring back todos 2026-06-01 15:48:10 +05:30
harisreedhar a88d8ead49 bring back todos 2026-06-01 15:45:44 +05:30
harisreedhar 7735c3740f bring back todos 2026-06-01 15:19:28 +05:30
henryruhs f2de9f1843 rename to video pack and audio pack 2026-06-01 11:41:25 +02:00
Harisreedhar aea36a9e55 Replace queues with timestamped deques and fix Audio Video sync (#1139)
* change to deque

* remove single line condition

* change threading to asyncio

* bringback todos
2026-06-01 15:03:33 +05:30
Henry Ruhs 0f5f75ba51 Cleanup/testing suite (#1136)
* clean testing suite

* clean testing suite part2

* clean testing suite part3

* add todos

* extend testing suite and kill some mutants

* fix hashes

* fix lint

* fix test

* fix test
2026-06-01 08:54:37 +02:00
Henry Ruhs 460c65004b Add available event (#1134)
* remove sleep with available event

* reorder methods, caller has to follow variable names of consumer, reorder tests methods

* more todos for naming
2026-05-30 16:39:48 +05:30
Henry Ruhs 1ac0e3e9a4 remove face store (#1132) (#1133) 2026-05-30 11:55:58 +02:00
Harisreedhar 9f6a19c1d2 feat(rtc): 2-way REMB with in-place encoder bitrate update (#1131)
* implement remb other direction

* update test
2026-05-30 04:41:10 +05:30
henryruhs 3ff327e670 add todos 2026-05-29 18:17:51 +02:00
henryruhs 2553ad7cad add todos 2026-05-29 18:10:15 +02:00
Harisreedhar 6b9ddd9a4f feat(rtc): REMB bitrate adaptation with in-place encoder update (#1130)
* improve test

* fix lint

* cleanup

* cleanup
2026-05-29 21:15:03 +05:30
henryruhs 871559cb6a clean and simplify tests 2026-05-29 15:44:07 +02:00
Harisreedhar 2a8672b54d Implement tier 1 REMB (#1129)
* implement tier 1

* fix lint

* cleanup

* cleanup

* fix lint

* use clear_remb method

* use single callback

* improve test

* improve test

* improve test

* improve test

* improve test
2026-05-29 19:00:27 +05:30
henryruhs cc0af9175a ban byte string 2026-05-26 19:15:11 +02:00
harisreedhar d77be89177 improve encoder collect 2026-05-26 22:25:55 +05:30
henryruhs e813d7df95 improve performance for decoder collect 2026-05-26 18:29:12 +02:00