search

lhwcv / mlsd_pytorch

Pytorch implementation of "M-LSD: Towards Light-weight and Real-time Line Segment Detection"
Apache License 2.0
190 stars 37 forks source link

some question about junction map #14

Open iifeve opened 2 years ago

iifeve commented 2 years ago

Hi there! Thanks for your excellent work, but I notice that the dimension of junction map is (h, w, 1), but the others are (c, h ,w), and in phase of loading label, it didn't change the dimension. I wonder should it be this way or there is something I missed?

 def gen_junction_and_line_mask(norm_lines, h = 256, w = 256):

junction_map = np.zeros((h, w, 1), dtype=np.float32) line_map = np.zeros((h, w, 1), dtype=np.float32)

def load_label(self, ann, do_aug):

     label[14, :, :] = junction_map[0]

label[15, :, :] = line_map[0]

ngdyes commented 1 year ago

Yeah,I found it wrong too.That makes line and junction segment effectless.