search

TimoSaemann / ENet

ENet: A Deep Neural Network Architecture for Real-Time Semantic Segmentation
577 stars 276 forks source link

Unsupported data layer type DenseImageData #43

Open Bardo91 opened 6 years ago

Bardo91 commented 6 years ago

I am trying to train the network with a custom dataset in DIGITs but I am getting the error "Unsupported data layer type DenseImageData".

Traceback (most recent call last):
  File "digits/scheduler.py", line 512, in run_task
    task.run(resources)
  File "digits/task.py", line 189, in run
    self.before_run()
  File "digits/model/tasks/caffe_train.py", line 220, in before_run
    self.save_files_generic()
  File "digits/model/tasks/caffe_train.py", line 632, in save_files_generic
    network = cleanedUpGenericNetwork(self.network)
  File "digits/model/tasks/caffe_train.py", line 1701, in cleanedUpGenericNetwork
    'Unsupported data layer type %s' % layer.type
AssertionError: Unsupported data layer type DenseImageData

I looked for it in the caffe-enet submodule and it is supposed to be registered on there in the cpp code. But it fails when I try to train the network

Thanks in advance.