plismbench.engine package

Top-level module for training and inference functionalities.

Subpackages

• plismbench.engine.extract package
  • Submodules
    • plismbench.engine.extract.core module
      • run_extract()
    • plismbench.engine.extract.extract_from_h5 module
      • H5Dataset
      • collate()
      • get_dataloader()
      • run_extract_h5()
    • plismbench.engine.extract.extract_from_png module
      • collate()
      • run_extract_streaming()
    • plismbench.engine.extract.utils module
      • sort_coords()
      • save_features()
      • process_imgs()

Submodules

• plismbench.engine.cli module
• plismbench.engine.evaluate module