IntelLabs / MART

Modular Adversarial Robustness Toolkit
BSD 3-Clause "New" or "Revised" License
16 stars 0 forks source link

Add YOLOv4 model #162

Open dxoigmn opened 1 year ago

dxoigmn commented 1 year ago

Some options:

I'm partial to the AlexeyAB repo.

dxoigmn commented 1 year ago

I veto https://github.com/Tianxiaomo/pytorch-YOLOv4 because it isn't torch-like due its use of self.inference state that smells a lot like self.training. That will cause all kinds of headaches.

dxoigmn commented 1 year ago

Interestingly, https://github.com/AlexeyAB/Yet-Another-YOLOv4-Pytorch implements "Self adversial training with fgsm": https://github.com/AlexeyAB/Yet-Another-YOLOv4-Pytorch/commit/ea3e0dc2c6d532de8d0ac342c0c9857a00056574