pyronear / pyro-vision

Computer vision library for wildfire detection 🌲 Deep learning models in PyTorch & ONNX for inference on edge devices (e.g. Raspberry Pi)
https://pyronear.org/pyro-vision/
Apache License 2.0
52 stars 24 forks source link

feat: Improved all model checkpoints and added HF hub loading function #159

Closed frgfm closed 2 years ago

frgfm commented 2 years ago

This PR introduces the following modifications:

Closes #123

Any feedback is welcome!

codecov[bot] commented 2 years ago

Codecov Report

Merging #159 (9442e99) into master (a3a73e5) will increase coverage by 0.04%. The diff coverage is 100.00%.

@@            Coverage Diff             @@
##           master     #159      +/-   ##
==========================================
+ Coverage   93.83%   93.88%   +0.04%     
==========================================
  Files           8        9       +1     
  Lines         211      229      +18     
==========================================
+ Hits          198      215      +17     
- Misses         13       14       +1     
Flag Coverage Δ
unittests 93.88% <100.00%> (+0.04%) :arrow_up:

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files Coverage Δ
pyrovision/models/mobilenetv3.py 100.00% <100.00%> (ø)
pyrovision/models/resnet.py 100.00% <100.00%> (ø)
pyrovision/models/rexnet.py 100.00% <100.00%> (ø)
pyrovision/models/utils.py 100.00% <100.00%> (ø)
pyrovision/datasets/utils.py 88.73% <0.00%> (-1.41%) :arrow_down:

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update a3a73e5...9442e99. Read the comment docs.