Unbatch ๐ฅ๐ ฅ๐ ๐ ข(Unbatch)
์ด ๋ ธ๋๋ ๊ธฐ์ ์ ์ผ๋ก PyTorch ํ ์(torch
Unbatch
Unbatch ๋ ธ๋๋ Video Helper Suite์ ์ ํธ๋ฆฌํฐ ๋๊ตฌ๋ก, ํต์ฌ ๊ธฐ๋ฅ์ ๋ฐฐ์น(batched)๋ ์ ๋ ฅ ๋ฐ์ดํฐ๋ฅผ ๋ถํดํ๊ฑฐ๋ ๋ณํฉํ์ฌ ๋จ์ผ ๋ฐ์ดํฐ ์คํธ๋ฆผ์ผ๋ก ๋ณต์ํ๋ ๊ฒ์ ๋๋ค. ์ฃผ๋ก ๋น๋์ค ๋๋ ์ด๋ฏธ์ง ์ํ์ค ์ฒ๋ฆฌ ๊ณผ์ ์์ ๋ฐฐ์น๋ก ํจํค์ง๋ ๋ฐ์ดํฐ๋ฅผ ์ฒ๋ฆฌํ์ฌ, ํ์์ ํ๋ ์๋ณ ๋๋ ๋ ๋ฆฝ์ ์ธ ์์ ์ ์ฉ์ดํ๊ฒ ํ๋ ๋ฐ ์ฌ์ฉ๋ฉ๋๋ค.
๋ ธ๋ ๊ธฐ๋ฅ
์ด ๋
ธ๋๋ ๊ธฐ์ ์ ์ผ๋ก PyTorch ํ
์(torch.Tensor), ํน์ ํค(์: 'samples' ๋๋ 'waveform')๋ฅผ ํฌํจํ๋ ๋์
๋๋ฆฌ, ๊ทธ๋ฆฌ๊ณ ๊ธฐํ ๋ฐ๋ณต ๊ฐ๋ฅํ ๊ฐ์ฒด๋ฅผ ํฌํจํ ๋ค์ํ ๋ฐ์ดํฐ ์ ํ์ ์ง์ํฉ๋๋ค. ํต์ฌ ๋ฉ์ปค๋์ฆ์ ์
๋ ฅ ๋ฐ์ดํฐ์ ์ฒซ ๋ฒ์งธ ์์ ์ ํ์ ํ์ธํ์ฌ ์ด๋ค ๋ณํฉ ์ ๋ต์ ์ฌ์ฉํ ์ง ๊ฒฐ์ ํฉ๋๋ค: ํ
์์ ๊ฒฝ์ฐ torch.cat์ ์ฌ์ฉํ์ฌ ์ฐ๊ฒฐํ๊ณ , ํน์ ๋์
๋๋ฆฌ์ ๊ฒฝ์ฐ ๋ด๋ถ ํ
์ ๋ฐ์ดํฐ๋ฅผ ๋ณํฉํ๋ฉฐ, ๋ค๋ฅธ ์ ํ์ ๊ฒฝ์ฐ ๋์ ์ฐ์ฐ์ ์ฌ์ฉํ์ฌ ๋ณํฉํฉ๋๋ค.
๋ ธ๋ ๋งค๊ฐ๋ณ์ ์ค๋ช - Unbatch ๐ฅ๐ ฅ๐ ๐ ข
์ฐ๊ฒฐ ์ ๋ ฅ (Inputs)
| ๋งค๊ฐ๋ณ์ ์ด๋ฆ | ๋ฐ์ดํฐ ์ ํ | ํ์ | ๊ธฐ๋ณธ๊ฐ | ๊ฐ ๋ฒ์/์ต์ | ์ค๋ช |
|---|---|---|---|---|---|
batched | * | ์ | - | - | ์ ๋ ฅ ๋งค๊ฐ๋ณ์ batched: ๋ฐฐ์น ์ฒ๋ฆฌ๋ ๋ฐ์ดํฐ ๋ชฉ๋ก์ ๋ฐ์, ๋ ธ๋๊ฐ ์ด๋ฅผ ์ฐ์์ ์ธ ๋ฐ์ดํฐ ์คํธ๋ฆผ์ผ๋ก ๋ณํฉํ๊ฑฐ๋ ๋ถํดํฉ๋๋ค. ์ ๋ ฅ์ ํน์ ๊ฐ ๋ฒ์ ์ ํ ์์ด ๋ชจ๋ ์ ํ์ ๋ชฉ๋ก์ด ๋ ์ ์์ต๋๋ค. |
์ฌ์ฉ ์๋๋ฆฌ์ค
์ผ๋ฐ์ ์ธ ๋น๋์ค ํ๋ ์ ํ์ฒ๋ฆฌ ์ํฌํ๋ก์์๋ ๋จผ์ Batch ๋ ธ๋๋ฅผ ์ฌ์ฉํ์ฌ ์ฌ๋ฌ ํ๋ ์ ์ด๋ฏธ์ง๋ฅผ ํจํค์งํ ํ, ์ผ๊ด ์ ์ค์ผ์ผ๋ง ๋๋ ํํฐ ์ฒ๋ฆฌ๋ฅผ ์ํํ ์ ์์ต๋๋ค. ์ฒ๋ฆฌ๊ฐ ์๋ฃ๋๋ฉด Unbatch ๋ ธ๋๋ฅผ ์ฌ์ฉํ์ฌ ์ด ๋ฐฐ์น ์ฒ๋ฆฌ๋ ์ด๋ฏธ์ง ํ๋ ์์ ๋ถ๋ฆฌํ์ฌ, ํ๋ ์๋ณ๋ก ์ ์ฅํ๊ฑฐ๋ ๋ ๋ฆฝ์ ์ธ ํ๋ ์ ๊ฐ ํฉ์ฑ ์์ ์ ์ํํ ์ ์์ต๋๋ค.
์ฃผ์์ฌํญ
์ด ๋
ธ๋๋ ์
๋ ฅ์ด ๋ชฉ๋ก ํํ์ ๋ฐ์ดํฐ์ฌ์ผ ํฉ๋๋ค(INPUT_IS_LIST = True๋ก ์ ์ด๋จ). ์ด๋ ์ผ๋ฐ์ ์ผ๋ก Batch ๋
ธ๋์ ๊ฐ์ด ๋ชฉ๋ก์ ์ถ๋ ฅํ๋ ๋
ธ๋๋ฅผ ์
๋ ฅ์ผ๋ก ์ฐ๊ฒฐํด์ผ ํจ์ ์๋ฏธํฉ๋๋ค. ๋์
๋๋ฆฌ ์ ํ์ ์
๋ ฅ์ ๋ํด์๋ ๋
ธ๋๊ฐ 'batch_index' ํค๊ฐ ์กด์ฌํ ๊ฒฝ์ฐ ์ด๋ฅผ ์ ๊ฑฐํฉ๋๋ค.
Unbatch ๐ฅ๐ ฅ๐ ๐ ข ๋ ธ๋ ์์ค ์ฝ๋ ๋งํฌ
Unbatch ๐ฅ๐ ฅ๐ ๐ ข ๋ ธ๋๋ ComfyUI-VideoHelperSuite ๋ ธ๋ ํจํค์ง์์ ๊ฐ์ ธ์์ต๋๋ค.
๋๊ธ
GitHub๋ก ๋ก๊ทธ์ธํ๊ณ ํ ๋ก ์ ์ฐธ์ฌํ์ธ์.