All pull requests

Pull requests

• align train_step with new object dict structure and loss input format

  #16·bernardev254 opened Jun 2, 2025·

  ·
  1

• complete evaluation by computing mAP on validation batches

  #15·bernardev254 opened Jun 2, 2025·

  ·
  1

• use eqx.combine and jax.vmap in loss_fn to compute detection outputs …

  #14·bernardev254 opened Jun 2, 2025·

  ·
  1

• implement YOLOv10-style loss function with bbox, objectness, and clas…

  #13·bernardev254 opened Jun 2, 2025·

  ·
  1

• add YOLOv10-style Equinox detection head for object detection outputs

  #12·bernardev254 opened Jun 2, 2025·

  ·
  1

• update augmentations to pass and return bbox and label transformation…

  #11·bernardev254 opened Jun 2, 2025·

  ·
  1

• apply crop and flip transformations to bboxes and labels

  #10·bernardev254 opened Jun 2, 2025·

  ·
  1

• implement preprocessing logic for object detection samples

  #9·bernardev254 opened Jun 1, 2025·

  ·
  1