ONNX, BCEBlurWithLogitsLoss, plot_study updates

models/onnx_export.py

@@ -1,6 +1,9 @@
-# Exports a pytorch *.pt model to *.onnx format
-# Example usage (run from ./yolov5 directory):
-# $ export PYTHONPATH="$PWD" && python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
+"""Exports a pytorch *.pt model to *.onnx format
+
+Usage:
+    import torch
+    $ export PYTHONPATH="$PWD" && python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
+"""
 
 import argparse