Boris Dayma, színező varázsló

dayma

A Colorizer Challenge

A texasi houstoni Boris Dayma az egyik bajnok volt a nyári színező versenyünkön. Ideghálózatot fejlesztett ki fekete-fehér képek készítésére és gyönyörű, színes megjelenítésekké alakítására. Szánjon egy percet az alábbi fekete-fehér és színes képek összehasonlítására.

Hogyan lehet megjósolni, hogy az egyes virágok milyen színűek lesznek? Ehhez kézzel kell kutatnia az egyes virágokat, és művelt találgatásokat kell tennie a csokor palettájáról és elrendezéséről. Amikor fekete-fehér filmeket színeznek, a művészek szorgalmasan képzelik el az egyes képkockák színeit, és egyedileg, kézzel festik a színt. Felhívtuk a kutatókat, hogy színesítsék a virágok fekete-fehér fotóit neurális hálózatokkal, és saját eredményeink nem voltak nagyok.

A színező számára nehéz meghatározni a jó veszteségfüggvényt, mert a jósolt és a megfelelő szín közötti távolság minimalizálásának egyszerű módja az, hogy kitalál valamit az összes szín közepén, ami végül barna lesz.

Mielőtt egy 2 hetes brazil nyaralásra indult, Boris kinyomtatott egy halom megjelent papírt színezékeken. Végiglapozta őket a repülőgépen, végigolvasta a tengerparti megvalósításokat, és fogalmat fogalmazott meg a probléma megközelítésének módjáról, hogy amikor visszaért az Egyesült Államokba, futásnak eredt.

Gondosan figyelemmel kísérte a folyamat edzésmodelljeit, a Weights & Biases valós idejű veszteséggörbéit használva azonosította a kiugró értékeket és korán megszakította az edzéseket, amikor nem teljesítettek jól.

Borisz módszere

A fekete-fehér képek alapértelmezés szerint az RGB színtérben vannak, ezért Boris áthelyezte a képeket az YCRCB térbe. Ez az egyik dimenziót csak a kép fényerejévé teszi, így leegyszerűsíti a problémát a CR és CB kiadásával. Saját építészetet épített, amelyet az U-Net, a MobileNets és a ResNet ihletett a képek szegmentálására. Borisz megtisztította az edzés adatait, és további képeket talált virágról, hogy kitöltse az edzéskészletet. Néhány adatbővítést is elvégzett - véletlenszerű kivágást és függőleges fordítást.

  • ‍Alapvonal - 5 réteg:Az első alapfutást 5 réteggel és 32 kezdeti szűrővel állítottuk be.
  • Alapvonal + feloldás: Az up-konvolúció használata az up-sampling helyett nem hozott javulást, és jelentősen megnövelte a modell méretét.
  • 6 réteg - súlyromlás:A súlycsökkenés használata túl lassú edzéshez vezet, még akkor is, ha a teljes veszteséghez való hozzájárulás többszörös csökkenése.
  • 6 réteg: A legjobb eredményeket 6 réteggel, 32 kezdeti szűrővel és szabályozás nélkül értük el.
    ‍

Eredmények

ResultsEredményei sokkal jobbak voltak, mint a szépia kimenetek, amelyeket a naiv modellünkkel kaptunk. Folyamatáról és eredményeiről többet megtudhat a Súlyok és elfogultságok projektjében. Nézze meg az eredménymintát - pontosan ki tudta képezni a modellt, hogy ezt a bogáncsvirágot lilának és a háttérfűnek zöldre színezze, nem könnyű teljesítmény!

Ünnepségünk

Örültünk az eredményeinek, és elrepítettük Borist, hogy találkozzunk a csapattal, és tegyünk egy kört Shivon Zilisszel. A délutánt fagylalt fogyasztásával töltötte, és megtapasztalta a Tesla autopilóta funkcióinak legújabb verzióját!