stesha2016 / lanenet-enet-hnet

Apache License 2.0
88 stars 33 forks source link

Is the weight file compact with original repo? #1

Closed vehxianfish closed 5 years ago

vehxianfish commented 5 years ago

I use original repo weight files and code python lanenet_hnet_predict.py --image_path ./data/tusimple_data/training/gt_image/0001.png --lanenet_weights ./model/tusimple_lanenet/tusimple_lanenet_vgg.ckpt ./model/hnet/hnet-2000 to predict single picture but some fault happens.
`

     [[Node: lanenet_model/inference/LaneNetBase/bottleneck1_0_main_max_pool =MaxPoolWithArgmax[T=DT_FLOAT, Targmax=DT_INT64, ksize=[1, 2, 2, 1], padding="SAME", strides=[1, 2, 2, 1]](lanenet_model/inference/LaneNetBase/iniatial_block_concat)]]

` it seems weights file is not match.If possible,can you share weight file for me.thank you!

stesha2016 commented 5 years ago

I have modified the backbone to ENet, so the original weights file will not match. Lanenet-ENet weights: 链接: https://pan.baidu.com/s/10rYFXoNDMW1RjNa3rvD9OA 提取码: qdt4 H-Net weights: 链接: https://pan.baidu.com/s/1n1ZMipCy-xDgum988H6q0w 提取码: ff2g

vehxianfish commented 5 years ago

thank u very much,that's cool!appreciation for your work!

stesha2016 commented 5 years ago

fixed