Add Dental Braces

Nota: Si raccomanda vivamente a tutti gli utenti di aggiornare alla versione 3 o 4. La versione 1 è di scarsa qualità.

Obiettivo:

Questo modello mira ad aggiungere apparecchi dentali fotorealistici ai denti dei soggetti nelle immagini generate. Gli apparecchi devono essere dimensionati e allineati con precisione per adattarsi naturalmente, rispondendo alle richieste che specificano il colore delle parentesi (es. oro o argento) e il colore delle bande.

Funzionalità attuali:

Non è perfetto ma la versione 4 funziona bene nella maggior parte dei casi. Funziona bene sia per maschi che femmine. Sebbene sia stato addestrato per aggiungere apparecchi fotorealistici a immagini fotorealistiche, durante i test ho scoperto che può aggiungere apparecchi praticamente a qualsiasi cosa, robot, animali, ecc. Ho inoltre riscontrato che funziona bene in altri stili, non solo foto, come disegni e stili cartoon.

Dati di addestramento:

La versione 4 di questo modello è stata addestrata su un set di 128 foto ad alta risoluzione (1024 x 1024) di apparecchi dentali reali, usando 6912 passi di addestramento attraverso 54 epoche. Il dataset consiste principalmente in immagini che mostrano diversi tipi di apparecchi e vari colori di bande.

Esempi di Prompt

Poiché il piano è che la LORA permetta di specificare il colore delle parentesi e del filo (es. oro o argento) e il materiale (plastica o metallo), ho iniziato ad aggiungere informazioni specifiche sul colore delle parentesi, filo e bande nelle etichette di ogni prompt.

Finora funziona solo il colore delle bande. In parte perché nel dataset attuale ci sono solo esempi di apparecchi con parentesi e bande argento.

Durante i test della versione 4, ho notato di ottenere risultati migliori e più coerenti se replicavo queste informazioni nel prompt.

Seguendo uno schema come:

adddentalbraces, un/a [Persona/Animale/Oggetto] [Età opzionale] [Se persona, mettere uomo o donna] con sui/sua/sue denti [colore] bande, parentesi argento e filo argento.

Esempi:

adddentalbraces, un uomo di 20 anni con apparecchio sui denti con bande rosse, parentesi argento e filo argento.

adddentalbraces, una donna di 20 anni con apparecchio sui denti con bande rosse, parentesi argento e filo argento.

Per chiarezza, NON è necessario promptare in questo modo, ma nei test ho ottenuto risultati più consistenti utilizzando questo stile, poiché i prompt sono più simili alle etichette del dataset e perché le etichette specificano il colore delle parentesi e del filo.

Risultati dei test A/B:

Versione 3 vs. Versione 4

Miglioramento della texture della pelle:

La versione 3 tendeva a rendere la pelle del volto con un aspetto troppo aerografato. Al contrario la versione 4 ha integrato immagini di regolarizzazione nel dataset per affrontare specificamente questo problema.

• I test A/B effettuati dopo l’addestramento hanno dimostrato un miglioramento sostanziale in questo ambito.

Dettagli migliorati del dataset:

La versione 4 include anche miglioramenti nel dataset per fornire rappresentazioni più dettagliate degli apparecchi reali.

• I test A/B hanno rivelato che la versione 4 ha introdotto occasionalmente caratteristiche sottili sugli apparecchi, come ganci per bande elastiche, assenti nella versione 3.

Rappresentazione di colori alternati delle bande:

La versione 4 comprende inoltre un numero maggiore di esempi con apparecchi che hanno colori delle bande alternati nel dataset.

• I test A/B hanno dimostrato che la versione 4 gestisce efficacemente prompt che specificano colori alternati delle bande (es. “con bande che alternano blu e verde”), superando la versione 3 sotto questo punto di vista.

Riepilogo delle prestazioni complessive:

In sintesi, la versione 4 ha prodotto risultati complessivamente leggermente migliori e meno anomalie, come sezioni delle parentesi che non dovrebbero esserci. Al contrario, la versione 3 talvolta produceva bande migliori ma a costo di una qualità generale dell’immagine inferiore.

Consigli per i migliori risultati

Usa descrittori specifici per età, genere e materiali degli apparecchi per migliorare il realismo.

Segui lo stile del prompt:

adddentalbraces, un/a [Persona/Animale/Oggetto] [Età opzionale] [Se persona, mettere uomo o donna] con sui/sua/sue denti [colore] bande, parentesi argento, e filo argento.

O almeno specifica un colore per le bande e usa la frase "con parentesi argento e filo argento."

Prompt semplici, come

adddentalbraces, una donna di 25 anni con apparecchio sui denti.

producono costantemente risultati molto soddisfacenti, generando foto in stile primo piano. Gli apparecchi risultano più estetici in questo stile ravvicinato e la LORA tende a creare immagini così, salvo diversa indicazione nel prompt.

La LORA può gestire anche prompt più complessi ed incorporare gli apparecchi con affidabilità, anche se nel caso di foto a figura intera o riprese a distanza la qualità degli apparecchi aggiunti può diminuire. Il dataset attuale manca di sufficienti esempi di apparecchi a distanza, portando a occasionali inconsistenze o rappresentazioni inaccurate. (Sto pianificando di includere ulteriori esempi di apparecchi a distanza nelle prossime versioni del dataset per migliorare questo aspetto, ma potrebbe richiedere tempo per ottenere o produrre immagini di qualità adeguata. Sono molto attento alla qualità delle immagini nel dataset e credo fermamente che la qualità dei dati sia il fattore principale per ottenere risultati di qualità dall’addestramento.)

Rimanete aggiornati! Feedback degli utenti sempre benvenuti, fatemi sapere cosa funziona e cosa necessita miglioramenti.