This is a port of the original implementation to PyTorch Lightning modules for extended flexibility. I have also added references to two example datasets I used for experimenting. The edge detection script has been extended by functionality to remove black borders around the target images for this dataset.
This is a port of the original implementation to PyTorch Lightning modules for extended flexibility. I have also added references to two example datasets I used for experimenting. The edge detection script has been extended by functionality to remove black borders around the target images for this dataset.