sleap icon indicating copy to clipboard operation
sleap copied to clipboard

Modify inference and centroid model to allow Bonsai ingestion

Open sheridana opened this issue 2 years ago • 1 comments

Description

  • Add function to InferenceModel that allows saving to a frozen graph, for ingestion into a bonsai pipeline
  • Add standalone centroid model

Types of changes

  • [ ] Bugfix
  • [x] New feature
  • [ ] Refactor / Code style update (no logical changes)
  • [ ] Build / CI changes
  • [ ] Documentation Update
  • [ ] Other (explain)

Does this address any currently open issues?

[list open issues here]

Outside contributors checklist

  • [ ] Review the guidelines for contributing to this repository
  • [ ] Read and sign the CLA and add yourself to the authors list
  • [ ] Make sure you are making a pull request against the develop branch (not main). Also you should start your branch off develop
  • [ ] Add tests that prove your fix is effective or that your feature works
  • [ ] Add necessary documentation (if appropriate)

Thank you for contributing to SLEAP!

:heart:

sheridana avatar Jul 21 '22 00:07 sheridana

Codecov Report

Merging #850 (a7a03ca) into develop (c4409dd) will increase coverage by 0.18%. The diff coverage is 90.58%.

@@             Coverage Diff             @@
##           develop     #850      +/-   ##
===========================================
+ Coverage    67.40%   67.59%   +0.18%     
===========================================
  Files          130      130              
  Lines        22110    22187      +77     
===========================================
+ Hits         14903    14997      +94     
+ Misses        7207     7190      -17     
Impacted Files Coverage Δ
sleap/nn/inference.py 79.54% <89.87%> (+4.35%) :arrow_up:
sleap/__init__.py 100.00% <100.00%> (ø)
sleap/nn/data/resizing.py 65.33% <100.00%> (-0.44%) :arrow_down:
sleap/nn/peak_finding.py 75.51% <100.00%> (-17.50%) :arrow_down:
sleap/nn/config/training_job.py 93.87% <0.00%> (+10.20%) :arrow_up:

:mega: We’re building smart automated test selection to slash your CI/CD build times. Learn more

codecov[bot] avatar Jul 21 '22 01:07 codecov[bot]