Glenn Jocher
5fac5ad165
Precision-Recall Curve Feature Addition ( #1107 )
* initial commit
* Update general.py
Indent update
* Update general.py
refactor duplicate code
* 200 dpi
4年前
Glenn Jocher
124f0e8212
torchvision nms bug fix
4年前
Glenn Jocher
883924d9dc
classifier, export, torch seed updates
4年前
Jiacong Fang
c5d2331897
Fix increment_dir to use run_xxx for logdir ( #1058 )
* Fix increment_dir to use run_xxx for logdir
Rerunning train.py with logdir in the form of `run_xxx` causes index slicing in
```
n = max([int(x[len(dir):x.find('_') if '_' in Path(x).name else None]) for x in d]) + 1 # increment
```
to underflow.
* Replace find with rfind
4年前
haoyuefan
7220cee1d1
mosaic4 bottom left image fix ( #1021 )
fix a bug in load_mosaic
4年前
Guilhen
702c4fa53e
adding the configuration to deploy yolov5 in in app engine ( #964 )
* add the configuration to deploy yolov5 in a docker container in app engine on google cloud
* remove the duplicated requirements, place the google app engine files in utils
* remove the unused requirements
Co-authored-by: guilhembau <guilhem@6dbytes.com>
4年前
Glenn Jocher
89655a84f2
.fuse() gradient introduction bug fix
4年前
Glenn Jocher
5a9c5c1d3b
add mosaic and warmup to hyperparameters ( #931 )
4年前
Glenn Jocher
f1c63e2784
add mosaic and warmup to hyperparameters ( #931 )
4年前
Glenn Jocher
806e75f2b1
self.label_files bug fix ( #958 )
4年前
Glenn Jocher
c54e39464c
check_file() assert unique
4年前
Glenn Jocher
720645d96a
LoadImagesAndLabels() /images/ to /labels/ fix ( #928 )
4年前
Glenn Jocher
d49c52eee3
_RepeatSampler outside of InfiniteDataLoader
4年前
Glenn Jocher
9c0efdde9f
glob.glob(recursive=True) search ( #917 )
4年前
Glenn Jocher
bc1fd13a7a
gsutil cp hyp evolution bug fix ( #876 )
4年前
Glenn Jocher
44cdcc7e0b
hyp['anchors'] evolution update
4年前
Glenn Jocher
ffe9eb4238
remove if platform.system() == 'Darwin' checks
4年前
tkianai
c5969f79f5
add support for DWConv fuse operation ( #898 )
* the value should be a list to unpacking
* filter the None labels
* update readme to remove apex dependency
* add support for DWConv fuse operation
4年前
Glenn Jocher
d3f9bf2bb7
Update datasets.py
4年前
NanoCode012
1e15aad6f9
Add InfiniteDataLoader class ( #876 )
* Add InfiniteDataLoader
Only initializes at first epoch. Saves time.
* Moved class to a better location
4年前
Glenn Jocher
987c226849
seaborn plotting correlogram addition
4年前
rafale77
9776e70988
torch.ops.torchvision.nms ( #860 )
Don't load the entire torchvision library just for nms when the function is already in the torch library.
4年前
Glenn Jocher
31d6c389f2
Update general.py plot_evolution()
4年前
Willie Maddox
455f7b8f76
Update general.py ( #823 )
Fixes #822
`init_seeds` from `torch_utils` import is being overwritten by function `init_seeds` in `general.py`
4年前
Glenn Jocher
5a7d79fbe6
Update general.py bbox_iou() with eps improvements ( #736 )
eps reduced to 1e-9 for additional precaution.
4年前
Glenn Jocher
82ed33a052
generalize webcam detection with opt.source.isnumeric() (yolov3 #1462 )
4年前
Glenn Jocher
4d7f222f73
Update export.py with v3.0 Hardswish() support
4年前
Glenn Jocher
8666bc507e
Create evolve.sh for infinite hyp evolution
4年前
Glenn Jocher
71209a6099
exportable Hardswish() implementation
4年前
Glenn Jocher
f7e075fe93
gsutil cp yaml when evolving to GCP bucket
4年前
Glenn Jocher
9dbaa1bbbe
Update plot_results()
4年前
Glenn Jocher
ea5e97fb9b
FP16 NMS ( #814 )
4年前
Glenn Jocher
b42e8a531b
optimize imports
4年前
Glenn Jocher
5e0b90de8f
CIoU nan bug fix ( #736 )
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
f5da528d28
reformat code
4年前
Quan Lin
66744a0df1
Update utils/google_utils.py ( #733 )
To download weights from v3.0 instead of v2.0
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年前
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年前
Marc
a925f283a7
max workers for dataloader ( #722 )
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年前
tkianai
9ae868364a
the value should be a list to unpacking ( #688 )
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年前
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年前
Glenn Jocher
3fcd365ba3
Fix curl download on Windows ( #669 )
4年前
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年前