Fully connected CNN which mimic enhancement techniques

Hello mates,
I’m thinking about building a Fully connected CNN which mimic an enhancement technique ( for example histogram equalisation) given the original input image.
Is there any techniques or model in pytorch could help me !?