Tf.image.total_variation pytorch implementation

Does pytorch have a implementation of TV??

Thanks

I don’t think there is a built-in total_variation method in torchvision, but based on the TF implementatio you could use try to use the same approach as it looks quite simple.

1 Like