yolov5/models
Glenn Jocher 19e2482458
PyTorch version to screen and cleanup (#1325)
* Create flatten_recursive() helper function

* cleanup

* print torch version
2020-11-09 12:24:11 +01:00
..
hub Update models/hub *.yaml files for v2.0 (#546) 2020-07-29 14:02:07 -07:00
__init__.py add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
common.py autoshape() update for PIL greyscale inputs (#1279) 2020-11-04 14:20:11 +01:00
experimental.py PyTorch 1.7.0 Compatibility Updates (#1233) 2020-10-28 15:03:50 +01:00
export.py Rearrange export input after checks (#1118) 2020-10-11 16:23:36 +02:00
yolo.py PyTorch version to screen and cleanup (#1325) 2020-11-09 12:24:11 +01:00
yolov5l.yaml update *.yaml output layer comments 2020-07-30 11:31:47 -07:00
yolov5m.yaml update *.yaml output layer comments 2020-07-30 11:31:47 -07:00
yolov5s.yaml update *.yaml output layer comments 2020-07-30 11:31:47 -07:00
yolov5x.yaml update *.yaml output layer comments 2020-07-30 11:31:47 -07:00