Hi, Before attempting to modify MNC to work with a 2-categories data

Is it possible to train an unmodified version on MNC with a different dataset? about mnc HOT 2 CLOSED

javierjsa commented on August 29, 2024

Is it possible to train an unmodified version on MNC with a different dataset?

from mnc.

Comments (2)

javierjsa commented on August 29, 2024

Some images in my dataset contained no objects, only background. Thus, their associated mat files had empty fields. Removing such files solved the issue. However, it crashes again after loading VGG16 caffe model, right before starting the training:

File "./tools/train_net.py", line 93, in <module>
    pretrained_model=args.pretrained_model)
  File "/MNC/tools/../lib/caffeWrapper/SolverWrapper.py", line 61, in __init__
    self.solver.net.layers[0].set_roidb(roidb)
  File "/MNC/tools/../lib/pylayer/mnc_data_layer.py", line 63, in set_roidb
    self._shuffle_roidb_inds()
  File "/MNC/tools/../lib/pylayer/mnc_data_layer.py", line 81, in _shuffle_roidb_inds
    inds = np.reshape(inds, (-1, 2))
  File "/usr/lib/python2.7/dist-packages/numpy/core/fromnumeric.py", line 218, in reshape
    return reshape(newshape, order=order)
ValueError: total size of new array must be unchanged

from mnc.

javierjsa commented on August 29, 2024

Setting C.TRAIN.ASPECT_GROUPING=False solved the problem, yet Caffe does not validate the model. So, in short, it can't be done

from mnc.

Recommend Projects