Code

Multi Input Support#36

Code

Merged

kozlov721 merged 16 commits intodevluxonis/luxonis-train:devfrom feature/multi-inputluxonis/luxonis-train:feature/multi-inputCopy head branch name to clipboardJun 13, 2024

+508-115

Commits

Commits on Jun 1, 2024

• Multi-input test case - building a complex multi-input POSET model with export

  CaptainTrojan

  committed
  4041c99

  Copy full SHA for 4041c99

• Added a test for new collate_fn, which collates dicts of tensors instead of bare tensors

  CaptainTrojan

  committed
  6809f98

  Copy full SHA for 6809f98

• Added a config for a multi-input model

  CaptainTrojan

  committed
  f400246

  Copy full SHA for f400246

• Added necessary changes to config: ModelNodeConfig has a parameter input_sources, which tells the LuxonisModel which loader sub-elements it wants to load, and LoaderConfig has an images_name parame…

  Show description for f546608
  CaptainTrojan

  committed
  f546608

  Copy full SHA for f546608

• Updated input_shape type and the return type of loader, the LuxonisLoaderTorchOutput.

  CaptainTrojan

  committed
  769aaab

  Copy full SHA for 769aaab

• Added images_name property to BaseNode class, removed redundant batch_dim from shapes

  CaptainTrojan

  committed
  9cf7ebb

  Copy full SHA for 9cf7ebb

• Implemented multi-input support in node building and model export

  CaptainTrojan

  committed
  4284fba

  Copy full SHA for 4284fba

• Compatibility changes due to the new way shapes work in loaders, making use of images_name.

  CaptainTrojan

  committed
  0e1ceff

  Copy full SHA for 0e1ceff

• [Automated] Updated coverage badge

  actions-user

  committed
  951b981

  Copy full SHA for 951b981

Commits on Jun 9, 2024

• Moved 'images_name' setting from loader implementation to config due to code structure requirements. Added missing tests for evaluation, export, and inference.

  CaptainTrojan

  committed
  1cdcb9f

  Copy full SHA for 1cdcb9f

Commits on Jun 12, 2024

• Merge branch 'dev' into feature/multi-input

  kozlov721

  committed
  6c69875

  Copy full SHA for 6c69875

• removed macos tests

  kozlov721

  committed
  6b07cd8

  Copy full SHA for 6b07cd8

Commits on Jun 13, 2024

• removed images_name from BaseNode

  kozlov721

  committed
  63c6ad0

  Copy full SHA for 63c6ad0

• renamed get_shape_pocket to to_shape_pocket

  kozlov721

  committed
  a4906e7

  Copy full SHA for a4906e7

• simplified input source handling

  kozlov721

  committed
  25d1758

  Copy full SHA for 25d1758

• renamed images_name to image_source

  kozlov721

  committed
  9987d0d

  Copy full SHA for 9987d0d