search

cardwing / Codes-for-Lane-Detection

Learning Lightweight Lane Detection CNNs by Self Attention Distillation (ICCV 2019)
MIT License
1.04k stars 333 forks source link

Training SCNN-Tensorflow model #302

Closed nisha1729 closed 4 years ago

nisha1729 commented 4 years ago

Hi, When I try to train the SCNN-Tensorflow model, I get the following error. Could you give some insight into what could be causing this problem? python=3.5, tensorflow=1.15.0, numpy=1.14.2

Traceback (most recent call last):
  File ".../Codes-for-Lane-Detection/SCNN-Tensorflow/lane-detection-model/tools/train_lanenet.py", line 335, in <module>
    train_net(args.dataset_dir, args.weights_path, net_flag=args.net)
  File ".../Codes-for-Lane-Detection/SCNN-Tensorflow/lane-detection-model/tools/train_lanenet.py", line 237, in train_net
    encoding='latin1').item()
  File "...\Python37\lib\site-packages\numpy\lib\npyio.py", line 440, in load
    pickle_kwargs=pickle_kwargs)
  File "...\Python37\lib\site-packages\numpy\lib\format.py", line 727, in read_array
    raise ValueError("Object arrays cannot be loaded when "
ValueError: Object arrays cannot be loaded when allow_pickle=False
nisha1729 commented 4 years ago

Turns out the versions are tensorflow=1.15.0 and python=3.7. Solved the error by setting allow_pickle = True in np.load() Also needs installing protobuf, thanks to https://github.com/cardwing/Codes-for-Lane-Detection/issues/121