Glenn Jocher
a8751e50de
model.fuse() fix for export.py ( #827 )
4年前
Glenn Jocher
5f07782f56
Update requirements.txt for coremltools kmeans quantization
4年前
Glenn Jocher
71209a6099
exportable Hardswish() implementation
4年前
Glenn Jocher
fd71fe8451
Update hyp.finetune.yaml
4年前
Glenn Jocher
f7e075fe93
gsutil cp yaml when evolving to GCP bucket
4年前
Glenn Jocher
9dbaa1bbbe
Update plot_results()
4年前
Glenn Jocher
83dc540b1d
remove ema.ema hasattr(ema, 'module') check
4年前
Glenn Jocher
1309400e6b
Update hyp.finetune.yaml on VOC results
4年前
Glenn Jocher
fb05681a04
Update requirements.txt to coremltools==4.0b3
4年前
Glenn Jocher
d0f98c0095
Update yolo.py with model_info(verbose=False)
4年前
Glenn Jocher
b9851222bf
Merge remote-tracking branch 'origin/master'
4年前
Glenn Jocher
c171e45cda
Update test.py with set() ( #806 )
4年前
Glenn Jocher
ea5e97fb9b
FP16 NMS ( #814 )
4年前
Glenn Jocher
b42e8a531b
optimize imports
4年前
Glenn Jocher
4447f4b937
--resume to same runs/exp directory ( #765 )
* initial commit
* add weight backup dir on resume
4年前
Glenn Jocher
5e0b90de8f
CIoU nan bug fix ( #736 )
4年前
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年前
Glenn Jocher
da35727c68
--save-txt bug fix ( #763 )
4年前
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年前
Francisco Ingham
26c3b11fdc
Update torch_utils.py ( #751 )
4年前
Glenn Jocher
2077d78880
update Dockerfile with libgl1-mesa-glx
4年前
Glenn Jocher
f5da528d28
reformat code
4年前
Glenn Jocher
b35f76e3d0
update .dockerignore
4年前
Glenn Jocher
fe80bed695
Update stale.yml
4年前
Glenn Jocher
7b2b52193d
update requirements.txt
4年前
Glenn Jocher
5991d144c1
Update README.md
4年前
Quan Lin
66744a0df1
Update utils/google_utils.py ( #733 )
To download weights from v3.0 instead of v2.0
4年前
Glenn Jocher
06bdd3955c
Update README.md
4年前
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年前
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年前
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年前
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年前
Marc
a925f283a7
max workers for dataloader ( #722 )
4年前
Glenn Jocher
7875f4c1fb
reverse plotting low to high confidence
4年前
Glenn Jocher
d2da523053
Merge remote-tracking branch 'origin/master'
4年前
Glenn Jocher
9f482cbcb8
utils.general comment updates/bug fixes
4年前
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年前
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年前
tkianai
542833c997
Filter the None labels ( #705 )
* the value should be a list to unpacking
* filter the None labels
4年前
Glenn Jocher
e71fd0ec0b
Model freeze capability ( #679 )
4年前
tkianai
9ae868364a
the value should be a list to unpacking ( #688 )
4年前
Glenn Jocher
8e5c66579b
update train.py remove save_json final_epoch
4年前
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年前
NanoCode012
3d8ed0a76b
Fix missing model.stride in DP and DDP mode ( #683 )
4年前
Glenn Jocher
a0ac5adb7b
Single-source training update ( #680 )
4年前
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年前
NanoCode012
d7cfbc47ab
Fix unrecognized local rank argument ( #676 )
4年前
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年前
Glenn Jocher
e742dcf417
update requirements.txt
4年前
Glenn Jocher
3fcd365ba3
Fix curl download on Windows ( #669 )
4年前