95 Commits

Author SHA1 Message Date
Glenn Jocher
4d3680c81d
Minor import and spelling updates (#1133) 2020-10-13 17:24:27 +02:00
Frederik
d11504aee1
Remove image open on macos (#1086)
* added option to not open on save

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-10-07 16:16:10 +02:00
stone99
b8f656742d
feat: add rtmp support (#1009)
* feat: add rtmp support

* Update detect.py

pass tuple to source.startswith()

Co-authored-by: ryan.fu <ryan.fu@aylaasia.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-09-22 11:37:49 -07:00
Glenn Jocher
82ed33a052 generalize webcam detection with opt.source.isnumeric() (yolov3 #1462) 2020-08-25 11:07:38 -07:00
Glenn Jocher
f5da528d28 reformat code 2020-08-14 11:53:44 -07:00
NanoCode012
0892c44bc4
Fix Logging (#719)
* Add logging setup

* Fix fusing layers message

* Fix logging does not have end

* Add logging

* Change logging to use logger

* Update yolo.py

I tried this in a cloned branch, and everything seems to work fine

* Update yolo.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-12 14:18:19 -07:00
Glenn Jocher
7875f4c1fb reverse plotting low to high confidence 2020-08-12 13:50:16 -07:00
Glenn Jocher
9f482cbcb8 utils.general comment updates/bug fixes 2020-08-11 20:11:44 -07:00
hopesala
d0d3dd1040
Update detect.py (#629)
* Update detect.py

Update print info for relative path and absolute
otherwise the info below is misleading

Results saved to /workspace//output_path

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-05 12:01:24 -07:00
lorenzomammana
728efa6576
Fix missing imports (#627)
* Fix missing imports

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-04 12:22:15 -07:00
Jirka Borovec
d5b6416c87
Explicit Imports (#498)
* expand imports

* optimize

* miss

* fix
2020-08-02 15:47:36 -07:00
Glenn Jocher
43a616a955 PyTorch 1.6.0 compatability updates 2020-07-28 15:38:56 -07:00
Glenn Jocher
fe094333e8
Update utils.py strip_optimizer() (#509)
Follow-on update that I missed adding into PR 509.
2020-07-24 19:30:37 -07:00
Glenn Jocher
02445d176d improved model.yaml source tracking 2020-07-13 14:35:47 -07:00
Glenn Jocher
78c9420e09 --classes bug fix #387 2020-07-13 13:27:46 -07:00
Glenn Jocher
094079bd26 update detect.py --class argument #387 2020-07-13 10:53:34 -07:00
Glenn Jocher
e8cf24b6c8 Initial model ensemble capability #318 2020-07-07 15:40:50 -07:00
Glenn Jocher
a40f615c6f .half() bug fix 2020-07-05 23:24:53 -07:00
Glenn Jocher
04bdbe4104 fuse update 2020-07-05 23:16:50 -07:00
Glenn Jocher
2541f77946 update detect.py 2020-07-05 11:57:48 -07:00
Glenn Jocher
1fdaa4987b update detect.py 2020-07-04 17:44:53 -07:00
Glenn Jocher
659ad748c8 update get_voc.sh 2020-07-04 17:13:43 -07:00
Glenn Jocher
5323ad224d export.py update 2020-07-01 16:15:25 -07:00
Glenn Jocher
a62a1c2c67 export.py update 2020-07-01 16:14:49 -07:00
Glenn Jocher
a9dc0c2c29 update check_img_size() for model strides 2020-06-30 15:25:38 -07:00
Russ Ferriday
0d5065d3b9
Remove non-ascii that breaks pdb single-stepping 2020-06-29 12:56:40 +01:00
Glenn Jocher
8c5b6220f5
Update detect.py 2020-06-25 19:26:23 -07:00
Glenn Jocher
496ec33a33
Update detect.py
Added some recent updates that were missing, and updated the filename with an if else.
2020-06-25 19:19:15 -07:00
Anish Hiranandani
68f63616b3 Fix after suggestion 2020-06-25 18:23:36 -07:00
Glenn Jocher
6c1b87a42e update google_utils import 2020-06-22 23:00:23 -07:00
Glenn Jocher
bbd12c7cfe detect.py fast bug fix 2020-06-22 10:14:58 -07:00
Lornatang
2368603484 fix refrence bug
In torch==1.5, the import of the API has changed. Although it does not interrupt the operation of the program, it seems to me to be an implicit error and may throw an exception in later versions.
2020-06-17 09:56:26 +08:00
Glenn Jocher
76ca367a01 FP16 to FP32 ckpt load 2020-06-15 13:32:45 -07:00
Glenn Jocher
111994956e switch default inference to FP16 on GPU 2020-06-14 19:25:39 -07:00
Glenn Jocher
260b1729f0 FP16 inference update 2020-06-14 18:32:29 -07:00
Glenn Jocher
099e6f5ebd --img-size stride-multiple verification 2020-06-12 22:10:46 -07:00
Glenn Jocher
ad71d2d513 update times to end-to-end latency (inference + NMS) 2020-06-12 11:42:08 -07:00
Glenn Jocher
c14368d768 new CSP model release 2020-06-09 14:31:04 -07:00
Glenn Jocher
c672bef10f model fuse 2020-06-07 14:07:22 -07:00
Glenn Jocher
dbdee3a4a3 multi-gpu test bug fix #15 2020-06-04 15:23:03 -07:00
Glenn Jocher
ba9ab663a2 model.names for dataParallel 2020-06-04 13:30:14 -07:00
Glenn Jocher
eb97b2e413 NMS fast mode 2020-06-03 13:02:59 -07:00
Glenn Jocher
391492ee5b updates 2020-06-02 23:57:58 -07:00
Glenn Jocher
df7988d8d0 onnx_export.py 2020-06-01 13:53:47 -07:00
Glenn Jocher
1e84a23f38 initial commit 2020-05-29 17:04:54 -07:00