Glenn Jocher
b9851222bf
Merge remote-tracking branch 'origin/master'
4 years ago
Glenn Jocher
c171e45cda
Update test.py with set() ( #806 )
4 years ago
Glenn Jocher
ea5e97fb9b
FP16 NMS ( #814 )
4 years ago
Glenn Jocher
b42e8a531b
optimize imports
4 years ago
Glenn Jocher
4447f4b937
--resume to same runs/exp directory ( #765 )
* initial commit
* add weight backup dir on resume
4 years ago
Glenn Jocher
5e0b90de8f
CIoU nan bug fix ( #736 )
4 years ago
NanoCode012
fb4fc8cd02
Fix ema attribute error in DDP mode ( #775 )
* Fix ema error in DDP mode
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 years ago
Glenn Jocher
da35727c68
--save-txt bug fix ( #763 )
4 years ago
Glenn Jocher
ebafd1ead5
single command --resume ( #756 )
* single command --resume
* else check files, remove TODO
* argparse.Namespace()
* tensorboard lr
* bug fix in get_latest_run()
4 years ago
Francisco Ingham
26c3b11fdc
Update torch_utils.py ( #751 )
4 years ago
Glenn Jocher
2077d78880
update Dockerfile with libgl1-mesa-glx
4 years ago
Glenn Jocher
f5da528d28
reformat code
4 years ago
Glenn Jocher
b35f76e3d0
update .dockerignore
4 years ago
Glenn Jocher
fe80bed695
Update stale.yml
4 years ago
Glenn Jocher
7b2b52193d
update requirements.txt
4 years ago
Glenn Jocher
5991d144c1
Update README.md
4 years ago
Quan Lin
66744a0df1
Update utils/google_utils.py ( #733 )
To download weights from v3.0 instead of v2.0
4 years ago
Glenn Jocher
06bdd3955c
Update README.md
4 years ago
Glenn Jocher
916d4aad9a
v3.0 Release ( #725 )
* initial commit
* remove yolov3-spp from test.py study
* update study --img range
* update mAP
* cleanup and speed updates
* update README plot
4 years ago
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
4 years ago
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>
4 years ago
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>
4 years ago
Marc
a925f283a7
max workers for dataloader ( #722 )
4 years ago
Glenn Jocher
7875f4c1fb
reverse plotting low to high confidence
4 years ago
Glenn Jocher
d2da523053
Merge remote-tracking branch 'origin/master'
4 years ago
Glenn Jocher
9f482cbcb8
utils.general comment updates/bug fixes
4 years ago
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>
4 years ago
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>
4 years ago
tkianai
542833c997
Filter the None labels ( #705 )
* the value should be a list to unpacking
* filter the None labels
4 years ago
Glenn Jocher
e71fd0ec0b
Model freeze capability ( #679 )
4 years ago
tkianai
9ae868364a
the value should be a list to unpacking ( #688 )
4 years ago
Glenn Jocher
8e5c66579b
update train.py remove save_json final_epoch
4 years ago
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
4 years ago
NanoCode012
3d8ed0a76b
Fix missing model.stride in DP and DDP mode ( #683 )
4 years ago
Glenn Jocher
a0ac5adb7b
Single-source training update ( #680 )
4 years ago
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
4 years ago
NanoCode012
d7cfbc47ab
Fix unrecognized local rank argument ( #676 )
4 years ago
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
4 years ago
Glenn Jocher
e742dcf417
update requirements.txt
4 years ago
Glenn Jocher
3fcd365ba3
Fix curl download on Windows ( #669 )
4 years ago
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>
4 years ago
Glenn Jocher
93684531c6
train.py --logdir argparser addition ( #660 )
* train.py --logdir argparser addition
* train.py --logdir argparser addition
4 years ago
NanoCode012
886b9841c8
Add Multi-Node support for DDP Training ( #504 )
* Add support for multi-node DDP
* Remove local_rank confusion
* Fix spacing
4 years ago
Glenn Jocher
7eaf225d55
zero-target training bug fix ( #609 )
4 years ago
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>
4 years ago
Glenn Jocher
b8f3b1b373
Update ci-testing.yml ( #637 )
* Update ci-testing.yml
* Update ci-testing.yml
4 years ago
lorenzomammana
728efa6576
Fix missing imports ( #627 )
* Fix missing imports
* Update detect.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 years ago
Glenn Jocher
941cb2680b
Merge remote-tracking branch 'origin/master'
4 years ago
Glenn Jocher
52b626392f
update README.md
4 years ago
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>
4 years ago