yolov5/models
Jean-Baptiste Martin 1cad0ce2c7
Allow `multi_label` option for NMS with PyTorch Hub (#4728)
* Allow specifying multi_label option for NMS when using torch hub

* Reformat

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-09-09 15:32:04 +02:00
..
hub Fix: add P2 layer 21 to yolov5-p2.yaml `Detect()` inputs (#4608) 2021-08-30 15:46:33 +02:00
__init__.py add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
common.py Allow `multi_label` option for NMS with PyTorch Hub (#4728) 2021-09-09 15:32:04 +02:00
experimental.py Add TensorFlow and TFLite export (#1127) 2021-08-17 13:18:16 +02:00
tf.py Add suffix checks (#4711) 2021-09-08 14:36:12 +02:00
yolo.py Add suffix checks (#4711) 2021-09-08 14:36:12 +02:00
yolov5l.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00
yolov5m.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00
yolov5s.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00
yolov5x.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00