Vincular codigo com foto

Ola boa tarde
Tem como vincular as fotos de um produto com os seus codigos, ou seja, eu tenho um produto que possuem algumas variações, entao criei um as categorias para ele, porem quando aparece no site o cliente clica na variação mas a foto nao muda. Nao sei se tem como fazer esse link Foto com o codigo.
No aguardo