@Skye25 great feature this but will need some customization. But if you wanna go the no code way, all you have to do is add featured images for all your variants in the product admin.
Then inside the customizer of your product template check the box to which the arrow points in the image below. This will show the particular image of the selected variant. Only limitation with this method is that you wont be able to show extra images for a particular color. This will show only the large image.
If this helps your case in any way please do like and accept it as a solution. Also if you need better customization or any related queries feel free to reach out at
Email ME
Buy Me A Coffee, If you feel i deserve it. Will be a motivating factor.
Thanks anyways.