Mishig Davaadorj
|
2a15e8ccfb
|
Object detection pipeline (#12886)
* Implement object-detection pipeline
* Define threshold const
* Add `threshold` argument
* Refactor
* Uncomment test inputs
* `rm
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Fix typo
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Fix typo
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Chore better doc
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Rm unnecessary lines
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Chore better naming
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Fix typo
* Add `detr-tiny` for tests
* Add `ObjectDetectionPipeline` to `trnsfrmrs/init`
* Implement new bbox format
* Update detr post_process
* Update `load_img` method obj det pipeline
* make style
* Implement new testing format for obj det pipeln
* Add guard pytorch specific code in pipeline
* Add doc
* Make pipeline_obj_tet tests deterministic
* Revert some changes to `post_process` COCO api
* Chore
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Rm timm requirement
* make fixup
* Add timm requirement to test
* Make fixup
* Guard torch.Tensor
* Chore
* Delete unnecessary comment
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
|
2021-09-08 17:17:32 +02:00 |
|