Glenn Jocher
8e9b3d0205
Precision-Recall Curve feature update #1206
пре 4 година
Glenn Jocher
ed85038c38
Precision-Recall Curve feature update ( #1206 )
* Precision-Recall Curve feature update
* sentinel value update
пре 4 година
Glenn Jocher
481d46cffb
Improved corruption handling during scan and cache ( #999 )
пре 4 година
Glenn Jocher
d61930e017
Improved corruption handling during scan and cache ( #999 )
пре 4 година
Shiwei Song
9291daa6f7
fix padding for rectangular inference ( #1165 )
Co-authored-by: swsong <swsong@stratosphere.mobi>
пре 4 година
Glenn Jocher
3b57cb5641
Simplified inference ( #1153 )
пре 4 година
Jirka Borovec
c67e72200e
fix compatibility for hyper config ( #1146 )
* fix/hyper
* Hyp giou check to train.py
* restore general.py
* train.py overwrite fix
* restore general.py and pep8 update
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
пре 4 година
Glenn Jocher
4d3680c81d
Minor import and spelling updates ( #1133 )
пре 4 година
Jirka Borovec
00917a6225
update expt name comment and folder parsing for training ( #978 )
* comment
* fix parsing
* fix evolve
* folder
* tqdm
* Update train.py
* Update train.py
* reinstate anchors into meta dict
anchor evolution is working correctly now
* reinstate logger
prefer the single line readout for concise logging, which helps simplify notebook and tutorials etc.
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
пре 4 година
Glenn Jocher
0ada058f63
Generalized regression criterion renaming ( #1120 )
пре 4 година
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 година