Kua whakaputaina e NVIDIA te waehere puna mo StyleGAN3, he punaha ako miihini e hangai ana i te whatunga neural adversarial (GAN) e whai ana ki te whakahiato i nga whakaahua pono o nga kanohi o te tangata. Ka tuhia te waehere ki te Python ma te whakamahi i te anga PyTorch ka tohatohahia i raro i te Raihana Waehere Puna NVIDIA, e whakatau ana i nga here mo te whakamahi arumoni.
Ko nga tauira kua oti te whakangungu kua oti te whakangungu i runga i te kohinga Flickr-Faces-HQ (FFHQ), kei roto 70 mano nga whakaahua o te kounga teitei (1024x1024) PNG o nga kanohi o te tangata, kei te waatea hoki mo te tango. I tua atu, he tauira i hangaia i runga i nga kohinga AFHQv2 (whakaahua o nga kanohi kararehe) me Metfaces (whakaahua o nga kanohi o te tangata mai i nga whakaahua o te peita tawhito). Ko te arotahi whanaketanga kei runga i nga kanohi, engari ka taea te whakangungu te punaha ki te whakaputa i nga mea katoa, penei i nga whenua me nga motuka. I tua atu, ka whakaratohia nga taputapu mo te whakangungu whaiaro i te whatunga neural ma te whakamahi i o kohinga whakaahua. Me hiahia kia kotahi, neke atu ranei nga kaari whakairoiro NVIDIA (Tesla V100, A100 GPU ranei e taunaki ana), 12 GB o RAM, PyTorch 1.9 me CUDA 11.1+ kete taputapu. Hei whakatau i te ahua horihori o nga kanohi ka puta, kei te hangaia he kaitirotiro motuhake.
Ma te punaha ka taea e koe te whakakotahi i tetahi ahua o te kanohi hou i runga i te whakakotahitanga o nga ahuatanga o te maha o nga kanohi, te whakakotahi i o raatau ahuatanga, me te whakarereke i te ahua whakamutunga ki te tau e hiahiatia ana, te ira tangata, te roa o nga makawe, te ahua ataata, te ahua o te ihu, tae kiri, mohiti, me te koki whakaahua. Ka whakaarohia e te kaihanga te ahua he kohinga o nga momo, ka wehe aunoa i nga taipitopito ahuatanga (te makawe, te makawe, te karaehe) mai i nga huanga taumata teitei noa (te ahua, te ira tangata, nga huringa tau) ka taea e koe te whakakotahi i a raatau i roto i tetahi ahua me te whakatau o te rangatira. āhuatanga mā te whakarea taumaha. Ko te mutunga mai, ka hangaia nga whakaahua e kore e rereke mai i nga whakaahua tuturu.
Ko te putanga tuatahi o te hangarau StyleGAN i whakaputaina i te tau 2019, i muri iho i tukuna he putanga pai ake o StyleGAN2020 i te tau 2, kia pai ake te ahua o te ahua me te whakakore i etahi taonga. I te wa ano, ka noho tonu te punaha, i.e. kare i whakaae kia tutuki te hākoritanga me te nekehanga kanohi. I te wa e whakawhanake ana i te StyleGAN3, ko te whainga matua ko te urutau i te hangarau mo tona whakamahinga i roto i te pakiwaituhi me te ataata.
Ka whakamahi a StyleGAN3 i te hoahoanga whakangao whakaahua hou, kaore he ingoa ingoa, me te whakatakoto i nga ahuatanga whakangungu whatunga neural hou. Kei roto i nga taputapu hou mo te whakaata tauwhitiwhiti (visualizer.py), tātaritanga (avg_spectra.py) me te whakaputa ataata (gen_video.py). Ko te whakatinanatanga ka whakaitihia te kohi mahara me te tere ake i te mahi ako.
Ko tetahi ahuatanga nui o te hoahoanga StyleGAN3 ko te whakawhiti ki te whakamaori i nga tohu katoa i roto i te whatunga neural i roto i te ahua o nga tukanga tonu, i taea ai, i te wa e hanga ana i nga waahanga, ki te whakahaere i nga waahi whanaungatanga kaore i te herea ki nga taunga tino o nga pika takitahi i roto te ahua, engari i mau ki te mata o nga mea e whakaatuhia ana. I roto i te StyleGAN me te StyleGAN2, ko te here ki nga pika i te wa o te whakatipuranga ka puta he raru i te wa o te whakaputanga hihiri, hei tauira, i te wa i neke ai te ahua, he rerekee nga korero iti, penei i nga kita me nga makawe, he ahua wehe ke atu i te toenga o te kanohi. . I StyleGAN3, kua whakatauhia enei raruraru, kua tino pai te hangarau mo te hanga ataata.
I tua atu, ka taea e tatou te kite i te panui o te hanganga a NVIDIA me Microsoft o te tauira reo nui rawa atu MT-NLG i runga i te whatunga neural hohonu me te hoahoanga "whakawhiti". Ka kapi te tauira i te 530 piriona tawhā, me te tautau o 4480 GPUs (560 DGX A100 tūmau me 8 A100 80GB GPU ia) i whakamahia mo te whakangungu. Ko nga tono a te tauira ko te whakaoti rapanga tukatuka reo maori, penei i te matapae i te otinga o nga rerenga korero kaore ano kia oti, te whakautu i nga paatai, te maarama ki te panui, te tuhi whakatau i te reo maori, me te whakakore i te tikanga o nga kupu.
Source: opennet.ru