Adicionar Aparelho Dentário

Nota: Todos os usuários são fortemente incentivados a atualizar para a versão 3 ou versão 4. A versão 1 é de baixa qualidade.

Objetivo:

Este modelo tem como objetivo adicionar aparelhos dentários foto-realistas aos dentes dos sujeitos em imagens geradas. Os aparelhos devem ter tamanho e alinhamento precisos para se encaixar naturalmente, respondendo a prompts que especificam a cor dos brackets (por exemplo, dourado ou prata) e a cor das bandas.

Funcionalidade Atual:

Não é perfeito, mas a versão 4 funciona e funciona bem na maioria das vezes. Funciona bem para homens e mulheres. Embora tenha sido treinado para adicionar aparelhos foto-realistas em imagens foto-realistas, nos testes descobri que pode adicionar aparelhos praticamente a qualquer coisa: robôs, animais, etc. Também descobri que funciona bem em outros estilos, não apenas fotos, como desenhos e estilos de desenhos animados.

Dados de Treinamento:

A versão 4 deste modelo foi treinada em um conjunto de 128 fotos em alta resolução (1024 x 1024) de aparelhos dentários reais, usando 6912 passos de treinamento ao longo de 54 épocas. O conjunto de dados consiste principalmente em imagens mostrando diferentes tipos de aparelhos e várias cores de bandas.

Exemplos de Prompts

Como o plano é que a LORA permita especificar a cor dos brackets e do fio (por exemplo, dourado ou prata) e o material (plástico ou metal), comecei a adicionar informações específicas sobre a cor dos brackets, do fio e das bandas nos rótulos de cada prompt.

Até agora, apenas a cor das bandas funciona. Isso em parte porque só há exemplos de aparelhos com brackets prateados e bandas prateadas no conjunto de dados atual.

Durante os testes da versão 4, percebi que obtinha resultados melhores e mais consistentes se espelhava essa informação no prompt.

Seguindo algo como:

adddentalbraces, um [Pessoa/Animal/Objeto] [Idade opcional] [Se for uma pessoa, coloque homem ou mulher] nos [seus/seu/da pessoa] dentes com bandas [cor], brackets prateados e um fio prateado.

Exemplos:

adddentalbraces, um homem de 20 anos com aparelho nos dentes com bandas vermelhas, brackets prateados e um fio prateado.

adddentalbraces, uma mulher de 20 anos com aparelho nos dentes com bandas vermelhas, brackets prateados e um fio prateado.

Para ficar claro, você NÃO precisa fazer o prompt dessa forma, mas nos testes obtive uma qualidade consistentemente melhor ao fazer assim, já que prompts nesse estilo são mais similares a muitos dos rótulos usados no conjunto de dados e porque os rótulos têm especificações como cor dos brackets e cor do fio.

Resultados dos Testes A/B:

Versão 3 vs. Versão 4

Melhoria na Textura da Pele:

A versão 3 apresentava uma tendência a criar uma aparência aerografada na pele do rosto. Em contraste, a versão 4 incorporou imagens de regularização em seu conjunto de dados para tratar especificamente essa questão.

• Testes A/B realizados após o treinamento demonstraram uma melhoria substancial nessa área.

Detalhes Aprimorados no Conjunto de Dados:

A versão 4 também traz melhorias no conjunto de dados para proporcionar representações mais detalhadas dos aparelhos reais.

• Testes A/B revelaram que a versão 4 ocasionalmente introduzia recursos sutis nos aparelhos, como ganchos para bandas elásticas, que não estavam presentes na versão 3.

Representação Alternada da Cor das Bandas:

A versão 4 também inclui uma gama maior de exemplos com aparelhos de cores alternadas nas bandas em seu conjunto de dados.

• Testes A/B demonstraram que a versão 4 lidava eficazmente com prompts que especificam as cores alternadas das bandas (por exemplo, “com bandas que alternam entre azul e verde”), superando a versão 3 nesse aspecto.

Resumo do Desempenho Geral:

Em resumo, a versão 4 atingiu resultados gerais ligeiramente superiores e apresentou menos anomalias, como seções do bracket que não deveriam estar presentes. Por outro lado, a versão 3 ocasionalmente produzia bandas superiores, mas à custa da qualidade geral da imagem.  

Dicas para Melhores Resultados

Use descritores específicos para idade, gênero e materiais do aparelho para melhorar o realismo.

Siga o estilo de prompt:

adddentalbraces, um [Pessoa/Animal/Objeto] [Idade opcional] [Se for uma pessoa, coloque homem ou mulher] nos [seus/seu/da pessoa] dentes com bandas [cor], brackets prateados e um fio prateado.

Ou ao menos especifique a cor das bandas e use a frase "com brackets prateados e um fio prateado."

Prompts simples, como

adddentalbraces, uma mulher de 25 anos com aparelho nos dentes.

geram consistentemente resultados altamente satisfatórios, produzindo fotografias estilo retrato. O aparelho é mais esteticamente agradável nesse estilo em close, e a LORA geralmente criará imagens nesse formato, a menos que o prompt peça explicitamente o contrário.

A LORA também pode lidar com prompts mais complexos e incorporar os aparelhos de forma confiável, embora se esses prompts resultarem em fotos de corpo inteiro ou à distância, a qualidade dos aparelhos adicionados pode diminuir. O conjunto de dados atual não possui exemplos suficientes de aparelhos à distância, levando a inconsistências ou representações imprecisas ocasionalmente. (Planejo incluir exemplos adicionais de aparelhos à distância na próxima versão do conjunto de dados para melhorar esse aspecto, mas isso pode levar tempo para obter ou produzir imagens de qualidade suficiente. Sou meticuloso com as imagens no conjunto de dados e acredito firmemente que a qualidade dos dados é o fator primordial para obter resultados de qualidade a partir do treinamento.)

Aguarde atualizações! Feedback dos usuários é sempre bem-vindo, me avise o que funciona e o que precisa melhorar.