yolov5

mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00

History

Add output_names argument for ONNX export with dynamic axes (#3456 )

* Add output names & dynamic axes for onnx export

Add output_names and dynamic_axes names for all outputs in torch.onnx.export. The first four outputs of the model will have names output0, output1, output2, output3

* use first output only + cleanup

Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

2021-06-04 22:01:58 +02:00

hub

Add TransformerLayer, TransformerBlock, C3TR modules (#2333 )

2021-04-01 17:26:53 +02:00

__init__.py

add "__init__.py" file in <models> folder

2020-06-15 14:37:07 +08:00

common.py

Parameterize max_det + inference default at 1000 (#3215 )

2021-05-17 22:47:44 +02:00

experimental.py

Enable direct --weights URL definition (#3373 )