SOLD2 output line and junction heatmaps have different shape than eachother and from the original tensor about kornia

[rpautrat]

Due to several downsampling and upsampling, the size of the heatmaps can indeed change. I think that the original dimensions must be divisible by 8 to keep the same resolution as in the output.

However, if I remember correctly, the output size will always be bigger than the input size, and you can simply crop the output to make it fit with the input one, removing the right and bottom borders. In your case, you could do this:

h, w = image_tensor.shape[2:4]
junction_heatmap_cropped = junction_heatmap[:, :h, :w]
line_heatmap_cropped = line_heatmap[:, :h, :w]

[edgarriba]

@ducha-aiki any insights here ?

[ducha-aiki]

@ogencoglu Actually, you are not supposed to use any of the heatmaps, the "official" output is line segments and descriptors :)
Anyway, let's tag authors here @rpautrat

[ogencoglu]

@ducha-aiki I know. But I need and am using heatmaps for other purposes as well.

I confirm that dims need to be divisible by 8 to have the same resolution. Thanks. This is enough I think.