Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
722 Commits 18 Branches 0 Tags
f5429260ca4fbc208fc9ba80804e7215fefc8f72
Commit Graph
55 Commits
Author SHA1 Message Date
Glenn Jocher f5429260ca PyTorch Hub and autoShape update (#1415) 
* PyTorch Hub and autoShape update

* comment x for imgs

* reduce comment
 2020-11-16 23:09:55 +01:00
Glenn Jocher fe341fa44d Utils reorganization (#1392) 
* Utils reorganization

* Add new utils files

* cleanup

* simplify

* reduce datasets.py

* remove evolve.sh

* loadWebcam cleanup
 2020-11-14 11:50:32 +01:00
Glenn Jocher c4addd7761 Unified '/project/name' results saving (#1377) 
* Project/name update

* Update ci-testing.yml

* address project with path separator failure mode

* Project/name update

* address project with path separator failure mode

* Update ci-testing.yml

* detect.py default --name bug fix

* missing rstrip PR

* train/exp0 to train/exp
 2020-11-12 23:37:46 +01:00
weida wangandGlenn Jocher 04081f8102 Trailing --save-txt whitespace bug fix (#1374) 
* update Write results

* Update detect.py

* Update test.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-11-12 17:37:16 +01:00
Khiem DoanandGlenn Jocher 1c8464e199 Use pathlib instead of low-level module (#1329) 
* Use pathlib instead of low-level module

* Use pathlib instead of low-level module

* Update detect.py

* Update test.py

* reformat

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-11-10 13:01:19 +01:00
Glenn Jocher 81d320109f --save-txt store_true fix (#1323) 2020-11-08 19:53:48 +01:00
Glenn Jocher 4821d076e2 Increment train, test, detect runs/ (#1322) 
* Increment train, test, detect runs/

* Update ci-testing.yml

* inference/images to data/images

* move images

* runs/exp to runs/train/exp

* update 'results saved to %s' str
 2020-11-08 19:39:05 +01:00
olegandGlenn Jocher 19c8b2c9b9 Adding --save-dir and --save-conf options to test.py (#1182) 
* Adding --output and --save-conf options to test.py

* Update help fields

* Update test.py

* Make arguments and comments uniform with test.py

* Remove previous and print save_dir on finish

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-10-25 16:50:21 +01:00
Osama AkhtarandGlenn Jocher 83deec133d Add confidence score to label txt files (#994) 
* add contrast to conf/class against bbox rectangle color

* added feature to save conf in txt labels

* Update general.py

* Update detect.py

* Update detect.py

* Update detect.py

save_conf variable removed as only used once, so we can use opt.save_conf instead.

* Update detect.py

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-10-20 14:20:27 +02:00
Glenn Jocher 3b57cb5641 Simplified inference (#1153) 2020-10-15 20:10:08 +02:00
Glenn Jocher 4d3680c81d Minor import and spelling updates (#1133) 2020-10-13 17:24:27 +02:00
FrederikandGlenn Jocher 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
stone99ryan.fuGlenn Jocher
 		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
NanoCode012andGlenn Jocher 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
hopesalaandGlenn Jocher 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
lorenzomammanaandGlenn Jocher 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