Yeah, that’s a good point and I’m really not sure which would work the best.
Just based on my gut feeling, I would assume both described alternatives might work better than going from the pre-trained 1000 class output to your custom layer.
Could you post your results in case you are trying some different approaches as this is quite interesting.