Uploaded image for project: 'Image Recognition'
  1. Image Recognition
  2. IMGREC-40

Train custom neural network for image recognition

XMLWordPrintable

    • Icon: Improvement Improvement
    • Resolution: Done
    • Icon: Neutral Neutral
    • None
    • None
    • None
    • Basel 161, Foundation 1
    • 13

      See MGNLPER-17 for background, user stories and business benefit.

      So far, we're using a provided model from the dl4j zoo with pre-trained weights for the ImageNet-1000 dataset, which is not a good fit for general-purpose image recognition (see MGNLPER-17 for details).

      Since there seem to be no pre-trained networks available that are a better fit, we should train our own. However, complete training from scratch should not be necessary; just doing transfer learning by "fine-tuning" an existing pre-trained model should be enough: Only replace the output layer with one that fits our new number of classes, then freeze all other layers and train on e.g. ImageNet data. Documentation about transfer learning with dl4j

      TBD: Classes (labels) we want to support

        Acceptance criteria

              creichenbach Cedric Reichenbach
              creichenbach Cedric Reichenbach
              Votes:
              0 Vote for this issue
              Watchers:
              3 Start watching this issue

                Created:
                Updated:
                Resolved:

                  Task DoD