Commit Graph

8 Commits

Author SHA1 Message Date
Gabriel Rasskin
a426717f10 Add IoU metrics: IoU, BinaryIoU, OneHotIoU, OneHotMeanIoU, (#127)
* Begin iou metrics

* Attempt conversion without confusion matrix backend

* Working ioumetrics, missing scatter op

* Formatting

* Docstring formatting

* Add IoU metrics to manifest

* Update with scatter op

* Fix scatter op for repeated indices

* Formatting

* Supress warning for core operation import

* Formatting
2023-05-15 20:50:55 -04:00
Francois Chollet
8c014ca995 Add Resizing layer and corresponding backend ops. 2023-05-05 17:12:09 -07:00
Chen Qian
841b8d702d Add max and average pooling layer (#66)
* Add max and poolig layer

* fix tests

* handle TF transpose

* renaming

* rename tests

* Fix comments

* Move out the shape computation logic
2023-05-02 15:17:59 -07:00
Ian Stenbit
a503324861 Add precision metric (#60)
* Add precision metric

* Implement tf nan_to_num

* Add epsilon to divisor

* Formatting
2023-04-30 10:10:07 -06:00
Haifeng Jin
ffa1f81b9e re-enable imports checks (#51) 2023-04-27 17:51:32 -07:00
Francois Chollet
14a10bcd1f Add bulk of serialization logic. 2023-04-21 10:00:32 -07:00
Chen Qian
eabdb87f9f Add some numpy ops (#1)
* Add numpy ops (initial batch) and some config

* Add unit test

* fix call

* Revert "fix call"

This reverts commit 6748ad183029ff4b97317b77ceed8661916bb9a0.

* full unit test coverage

* fix setup.py
2023-04-12 11:31:58 -07:00
Francois Chollet
160473c474 Initial draft of the codebase. 2023-04-09 12:21:45 -07:00