Commit Graph
1 Commits
Author SHA1 Message Date
Andrei Ionut DamianandGlenn Jocher 938efe5dcf update detect.py in order to support torch script (#5109)
* update detect.py in order to support torch script

This change assumes the torchscrip file was previously saved with `export.py`

* update `detect.py` for torchscript support

Simple update for torchscript support. Assumes the torchscript file has been generated with `export.py`

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-10-11 21:56:06 -07:00