Open dxoigmn opened 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.
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
Some options:
I'm partial to the AlexeyAB repo.