Skip to content

v2.0.0

Choose a tag to compare

@mzur mzur released this 02 Feb 09:28

This module now uses PyTorch and MMDetection instead of TensorFlow. The MAIA object detection is now implemented with Faster R-CNN instead of Mask R-CNN. In a few tests, this new setup performed a little better than before (higher recall and precision). The results were a lot more stable, too (i.e. recall and precision did not change that much when training/inference was repeated on the same data).

Upgrade information

The MAIA_AVAILABLE_BYTES env variable was removed. Instead, the object detection training batch size can be set directly with MAIA_MMDET_TRAIN_BATCH_SIZE. The default batch size is 16 which should work with about 12 GB of GPU memory.

The COCO_MODEL_URL env variable was removed. Use MAIA_BACKBONE_MODEL_URL for the PyTorch backbone file and MAIA_MODEL_URL for the MMDetection checkpoint file.

The training scheme is no longer configurable. Instead of iteration-based training (which still was called "epochs" before), training runs for a fixed number of 12 epochs now (which is the default training config for many MMDet object detectors and worked well in tests).

What's previously been called "instance segmentation" is now correctly called "object detection".

If you want to upgrade your BIIGLE production instance based on biigle/biigle:gpu, please pull the latest changes and also update the build/.env file.

What's Changed

Full Changelog: v1.24.0...v2.0.0