search

Trusted-AI / adversarial-robustness-toolbox

Adversarial Robustness Toolbox (ART) - Python Library for Machine Learning Security - Evasion, Poisoning, Extraction, Inference - Red and Blue Teams
https://adversarial-robustness-toolbox.readthedocs.io/en/latest/
MIT License
4.76k stars 1.15k forks source link

YOLO Object Detection Estimator for TensorFlow #1996

Open f4str opened 1 year ago

f4str commented 1 year ago

Is your feature request related to a problem? Please describe. The current YOLO object detection estimator only has an implementation for PyTorch. There is no YOLO estimator for TensorFlow.

Describe the solution you'd like An implementation of the YOLO object detection estimator for TensorFlow similar to the current PyTorch implementation. This should be located in art.estimators.object_detection.pytorch_object_detector and have the class name TensorFlowV2Yolo following the current conventions.

Describe alternatives you've considered N/A

Additional context N/A

beat-buesser commented 1 year ago

Hi @f4str This sounds good. Which versions of YOLO do you plan to support?