yolov5_ZWgra

0 Fork 0

Files

T

Jebastin NadarandGlenn Jocher a0e15046c3 Fix different devices bug when moving model from GPU to CPU (#5110 )

* fix different devices bug

* extend _apply() instead of to() for a general fix

* Only apply if Detect() is last layer

Co-authored-by: Jebastin Nadar <njebastin10@gmail.com>

* Indent fix

* Add comment to yolo.py

* Add comment to common.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

2021-10-10 14:24:05 -07: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

Fix different devices bug when moving model from GPU to CPU (#5110 )

2021-10-10 14:24:05 -07:00

experimental.py

Add TensorFlow and TFLite export (#1127 )