Glenn Jocher
|
d3f9bf2bb7
|
Update datasets.py
|
4 years ago |
NanoCode012
|
901243c780
|
Fix relative path for VOC download (#880)
Cleaned up code
|
4 years ago |
Glenn Jocher
|
3e04d20c7d
|
Merge remote-tracking branch 'origin/master'
|
4 years ago |
Glenn Jocher
|
f06e2d518c
|
opt.image_weights bug fix (#885)
|
4 years ago |
NanoCode012
|
1e15aad6f9
|
Add InfiniteDataLoader class (#876)
* Add InfiniteDataLoader
Only initializes at first epoch. Saves time.
* Moved class to a better location
|
4 years ago |
Glenn Jocher
|
69ff781ca5
|
opt.img_weights bug fix (#885)
|
4 years ago |
Glenn Jocher
|
987c226849
|
seaborn plotting correlogram addition
|
4 years ago |
Glenn Jocher
|
08e97a2f88
|
Update hyperparameters to add lrf, anchors
|
4 years ago |
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 years ago |
Glenn Jocher
|
0c01afc390
|
Update Dockerfile to upgrade pip
|
4 years ago |
Glenn Jocher
|
62e037419f
|
Update Dockerfile FROM nvcr.io/nvidia/pytorch:20.08-py3
|
4 years ago |
Glenn Jocher
|
31d6c389f2
|
Update general.py plot_evolution()
|
4 years ago |
Glenn Jocher
|
1e7674d4dd
|
Update export.py --img-size comment
|
4 years ago |
Glenn Jocher
|
a21bd0687c
|
Update train.py forward simplification
|
4 years ago |
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 years ago |
Glenn Jocher
|
5a7d79fbe6
|
Update general.py bbox_iou() with eps improvements (#736)
eps reduced to 1e-9 for additional precaution.
|
4 years ago |
Glenn Jocher
|
82ed33a052
|
generalize webcam detection with opt.source.isnumeric() (yolov3 #1462)
|
4 years ago |
Glenn Jocher
|
09402a2174
|
torch.from_tensor() bug fix
|
4 years ago |
Glenn Jocher
|
4fb8cb353f
|
Update export.py with v3.0 Hardswish() support (#831)
|
4 years ago |
Glenn Jocher
|
4d7f222f73
|
Update export.py with v3.0 Hardswish() support
|
4 years ago |
Glenn Jocher
|
8666bc507e
|
Create evolve.sh for infinite hyp evolution
|
4 years ago |
Glenn Jocher
|
a8751e50de
|
model.fuse() fix for export.py (#827)
|
4 years ago |
Glenn Jocher
|
5f07782f56
|
Update requirements.txt for coremltools kmeans quantization
|
4 years ago |
Glenn Jocher
|
71209a6099
|
exportable Hardswish() implementation
|
4 years ago |
Glenn Jocher
|
fd71fe8451
|
Update hyp.finetune.yaml
|
4 years ago |
Glenn Jocher
|
f7e075fe93
|
gsutil cp yaml when evolving to GCP bucket
|
4 years ago |
Glenn Jocher
|
9dbaa1bbbe
|
Update plot_results()
|
4 years ago |
Glenn Jocher
|
83dc540b1d
|
remove ema.ema hasattr(ema, 'module') check
|
4 years ago |
Glenn Jocher
|
1309400e6b
|
Update hyp.finetune.yaml on VOC results
|
4 years ago |
Glenn Jocher
|
fb05681a04
|
Update requirements.txt to coremltools==4.0b3
|
4 years ago |
Glenn Jocher
|
d0f98c0095
|
Update yolo.py with model_info(verbose=False)
|
4 years ago |
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 |