If you really want then just tag them, but building with a cache would be very quick when you swap.
> does the state of my docker container match the Dockerfile of my feature branch"
Why track the state of your docker container? Just build it.