Commit Graph

561 Commits (1ddf692bf89ac47d5e2dc8333cd0f4f145bb7f61)
 

Author SHA1 Message Date
tkianai 1ddf692bf8
Update readme to remove apex dependency (#727)
* the value should be a list to unpacking

* filter the None labels

* update readme to remove apex dependency
2020-08-12 21:18:29 -07:00
Hatovix 56c2c344ff
Fix list paths (#721)
* Add list paths on check_dataset

* missing raise statement

* Update general.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-12 18:53:38 -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
Marc a925f283a7
max workers for dataloader (#722) 2020-08-12 13:57:36 -07:00
Glenn Jocher 7875f4c1fb reverse plotting low to high confidence 2020-08-12 13:50:16 -07:00
Glenn Jocher d2da523053 Merge remote-tracking branch 'origin/master' 2020-08-11 20:11:58 -07:00
Glenn Jocher 9f482cbcb8 utils.general comment updates/bug fixes 2020-08-11 20:11:44 -07:00
Marc 1f92422e20
Github release assets model autodownload (#711)
* assets autodownload

* Update google_utils.py

* Update google_utils.py

* Update google_utils.py

* Update google_utils.py

* Update google_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-11 17:17:24 -07:00
NanoCode012 4949401a94
Fix redundant outputs via Logging in DDP training (#500)
* Change print to logging

* Clean function set_logging

* Add line spacing

* Change leftover prints to log

* Fix scanning labels output

* Fix rank naming

* Change leftover print to logging

* Reorganized DDP variables

* Fix type error

* Make quotes consistent

* Fix spelling

* Clean function call

* Add line spacing

* Update datasets.py

* Update train.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-11 11:18:45 -07:00
tkianai 542833c997
Filter the None labels (#705)
* the value should be a list to unpacking

* filter the None labels
2020-08-11 11:15:48 -07:00
Glenn Jocher e71fd0ec0b Model freeze capability (#679) 2020-08-10 22:49:43 -07:00
tkianai 9ae868364a
the value should be a list to unpacking (#688) 2020-08-10 11:23:48 -07:00
Glenn Jocher 8e5c66579b update train.py remove save_json final_epoch 2020-08-09 21:24:40 -07:00
Glenn Jocher 41523e2c91
Dataset autodownload feature addition (#685)
* initial commit

* move download scripts into data/scripts

* new check_dataset() function in general.py

* move check_dataset() out of with context

* Update general.py

* DDP update

* Update general.py
2020-08-09 20:52:57 -07:00
NanoCode012 3d8ed0a76b
Fix missing model.stride in DP and DDP mode (#683) 2020-08-09 11:01:36 -07:00
Glenn Jocher a0ac5adb7b Single-source training update (#680) 2020-08-09 02:27:35 -07:00
Glenn Jocher 3c6e2f7668
Single-source training (#680)
* Single-source training

* Extract hyperparameters into seperate files

* weight decay scientific notation yaml reader bug fix

* remove import glob

* intersect_dicts() implementation

* 'or' bug fix

* .to(device) bug fix
2020-08-09 02:12:44 -07:00
NanoCode012 d7cfbc47ab
Fix unrecognized local rank argument (#676) 2020-08-08 16:40:10 -07:00
Glenn Jocher 0e207551aa
update greetings.yml with pull_request_target (#677)
Should resolve "Resource not accessible by integration" error message.

https://github.com/actions/first-interaction/issues/10#issuecomment-670968624
2020-08-08 15:35:40 -07:00
Glenn Jocher e742dcf417 update requirements.txt 2020-08-08 12:57:05 -07:00
Glenn Jocher 3fcd365ba3 Fix curl download on Windows (#669) 2020-08-07 12:13:55 -07:00
NanoCode012 fe5b3f8712
Fix curl download on Windows (#669)
* Update ci-testing.yml

* Fix windows download

* Fix cookie and curl download issue from gdrive

* Revert "Update ci-testing.yml"

This reverts commit 7389d2238d.

* Update google_utils.py

PEP8

* Update google_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-07 12:10:59 -07:00
Glenn Jocher 93684531c6
train.py --logdir argparser addition (#660)
* train.py --logdir argparser addition

* train.py --logdir argparser addition
2020-08-06 22:26:38 -07:00
NanoCode012 886b9841c8
Add Multi-Node support for DDP Training (#504)
* Add support for multi-node DDP

* Remove local_rank confusion

* Fix spacing
2020-08-06 11:15:24 -07:00
Glenn Jocher 7eaf225d55 zero-target training bug fix (#609) 2020-08-05 13:35:31 -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
Glenn Jocher b8f3b1b373
Update ci-testing.yml (#637)
* Update ci-testing.yml

* Update ci-testing.yml
2020-08-05 11:42:51 -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
Glenn Jocher 941cb2680b Merge remote-tracking branch 'origin/master' 2020-08-03 19:55:57 -07:00
Glenn Jocher 52b626392f update README.md 2020-08-03 19:55:12 -07:00
Wang Xinyu 7ecf09d6da
Update README.md, deploy model with tensorrt (#623)
* Update README.md, deploy model with tensorrt

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-03 19:41:50 -07:00
Glenn Jocher 4ee8adb04b Merge remote-tracking branch 'origin/master' 2020-08-03 17:18:37 -07:00
Glenn Jocher f346da9f2b update activations.py 2020-08-03 17:18:31 -07:00
Glenn Jocher f807e7b867
Update README.md 2020-08-03 11:11:12 -07:00
Glenn Jocher 937b9db977
Update README.md 2020-08-03 11:09:46 -07:00
Glenn Jocher eb99dff9ef import random bug fix (#614) 2020-08-03 10:52:59 -07:00
Jirka Borovec d5b6416c87
Explicit Imports (#498)
* expand imports

* optimize

* miss

* fix
2020-08-02 15:47:36 -07:00
Glenn Jocher ec7a926163 AutoAnchor update to display anchors/target 2020-08-02 14:38:45 -07:00
Glenn Jocher d989bc9260 remove NBSP 2020-08-02 14:23:05 -07:00
Glenn Jocher 023e37807c hyperparameter evolution update (#566) 2020-08-02 13:18:03 -07:00
Glenn Jocher 64e6d199ba hyperparameter evolution update (#566) 2020-08-02 13:17:26 -07:00
Glenn Jocher fdbcc8fd7f hyperparameter evolution update (#566) 2020-08-02 12:30:34 -07:00
Glenn Jocher f1096b2cf7 hyperparameter evolution update (#566) 2020-08-02 10:32:04 -07:00
Glenn Jocher 305c6a028a compute_loss() leaf variable update 2020-08-02 10:30:43 -07:00
Glenn Jocher c1a2a7a411 hyperparameter evolution bug fix (#566) 2020-08-01 23:00:10 -07:00
Glenn Jocher 8074745908 hyperparameter evolution bug fix (#566) 2020-08-01 19:24:14 -07:00
Glenn Jocher e32abb5fb9 hyperparameter evolution bug fix (#566) 2020-08-01 19:15:48 -07:00
Glenn Jocher 8056fe2db8 hyperparameter evolution bug fix (#566) 2020-08-01 15:07:40 -07:00
Glenn Jocher 61b5733c36 update requirements.txt 2020-08-01 14:00:38 -07:00
Glenn Jocher 3e5e44499d update requirements.txt 2020-08-01 14:00:14 -07:00