-
Notifications
You must be signed in to change notification settings - Fork 10
[OTE / XAI] Handle two stage detector in the inferrer.py #104
Conversation
* add simple_test two-stage * two-path for saliency map exporting * two-stage detector exception for simple test * remove assertion for class-wise saliency map * fix assertion error in the tiling det
…toolkit/model_preparation_algorithm into dk/fix_det_saliency_map
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please redirect to releases/OTE_0.4.0 branch
Could you add a test to reproduce this bug? We couldn't find this in OTE CI. |
@goodsong81 Yes, I've changed the branch to release.
Yes, seems strange why it didn't occur in the OTE test.. I'll also check and add it. |
01b3f45
to
381c3b0
Compare
Thank you for your hard work! |
@harimkang @goodsong81 |
@harimkang @goodsong81 The reason why the test code has passed is currently |
It would be nice if tests |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Changes look good to me but could you create OTE PR w/ some tests for this PR and run CI tests?
@harimkang @goodsong81 Yes, I've made a PR to check from a single sample saliency map exporting step, though it could be improved in the test_xai improvement PR |
But we have some task.infer API tests https://github.com/openvinotoolkit/training_extensions/blob/46b831bcdb7db8e3dc08e819c37969de55c53b4d/external/model-preparation-algorithm/tests/api_tests/test_ote_detection_api.py#L221-L253 that actually run inference with To test this PR we need task.infer API tests for two-stage detectors, which we do not have running for now, as I can see. Or am I missing it? |
Yes, that's right. I think our test is missing the API test for Mask-RCNN (2 stage detector). |
Let's handle missing tests in another PRs. I'm merging this PR. |
I might be mistaken, but aren't we testing the two-stage detectors in the |
* [XAI] hot-fix of error in Detection XAI support (#99) * add simple_test two-stage * two-path for saliency map exporting * two-stage detector exception for simple test * remove assertion for class-wise saliency map * fix assertion error in the tiling det * [OTE / XAI] Handle two stage detector in the inferrer.py (#104) * [XAI] hot-fix of error in Detection XAI support (#98) * add simple_test two-stage * two-path for saliency map exporting * two-stage detector exception for simple test * remove assertion for class-wise saliency map * fix assertion error in the tiling det * handle two stage detectorin the inferrer.py * handle two stage detectorin the inferrer.py * handle two stage detectorin the inferrer.py * handle from detector instance checking * accept tuple data * [Hot-fix] Fix zero-division error in one cycle lr scheduler in multilabel classification Co-authored-by: dongkwan-kim <[email protected]> Co-authored-by: Soobee Lee <[email protected]>
* fixed multilabel configs (#67) * Tiling Module (#40) * Feature/val batch and seed (#69) * workaround bug (#70) * Kp/devide runners (#71) * move ema model to hook (#73) * Appley release/OTE_0.3.0 changes (#77) * Per-group softmax output (#74) * Move drop_last in cls trainer.py (#79) * Removed unnecessary mim version constraint for networkx package (#80) * Revert "Removed unnecessary mim version constraint for networkx package (#80)" (#82) * Don't apply activations on export in classification (#83) * Delete unused code (#84) * Replace current saliency map generation with Recipro-CAM for cls (#81) * Class-wise saliency map generation for the detection task (#97) * [XAI] hot-fix of error in Detection XAI support (#98) * [XAI] hot-fix of error in Detection XAI support (#99) * Replace only the first occurrence in the state_dict keys (#91) * [OTE / XAI] Handle two stage detector in the inferrer.py (#104) * [XAI] hot-fix of error in Detection XAI support (#98) * [OTE / XAI][Develop] Handle two stage detector in the inferrer.py (#107) * [XAI] hot-fix of error in Detection XAI support (#99) * Merge OTE side XAI update to OTX (#109) * Merge back releases/OTE_0.4.0 to develop (#116) * [XAI] hot-fix of error in Detection XAI support (#99) * [OTE / XAI] Handle two stage detector in the inferrer.py (#104) * [XAI] hot-fix of error in Detection XAI support (#98) * [Hot-fix] Fix zero-division error in one cycle lr scheduler in multilabel classification * Fix extra activations when exporting nonlinear hierarchical head (#118) * Fix get_train_data_cfg -> get_data_cfg Signed-off-by: Songki Choi <[email protected]> Co-authored-by: Prokofiev Kirill <[email protected]> Co-authored-by: Eugene Liu <[email protected]> Co-authored-by: Sungman Cho <[email protected]> Co-authored-by: Jihwan Eom <[email protected]> Co-authored-by: Yunchu Lee <[email protected]> Co-authored-by: Vladislav Sovrasov <[email protected]> Co-authored-by: Evgeny Tsykunov <[email protected]> Co-authored-by: dongkwan-kim <[email protected]> Co-authored-by: Adria Arrufat <[email protected]> Co-authored-by: Soobee Lee <[email protected]>
This is the hot-fix while running the Geti side OTE inference, fix handling the detection saliency map hook in the
inferrer.py
.