Glenn Jocher
23a7f29e2d
Update hyp.finetune.yaml on 306 VOC generations result
před 4 roky
Glenn Jocher
0a0675da12
Merge remote-tracking branch 'origin/master'
před 4 roky
Glenn Jocher
354a97e4ef
Create sotabench.py
před 4 roky
NanoCode012
d8274d0434
Fix results_file not renaming ( #903 )
před 4 roky
Glenn Jocher
281d78c105
Update train.py ( #902 )
* Update train.py with simplified ckpt names
* Return default hyps to hyp.scratch.yaml
Leave line commented for future use once mystery of best finetuning hyps to apply becomes clearer.
* Force test_batch*_pred.jpg replot on final epoch
This will allow you to see predictions final testing run after training completes in runs/exp0
před 4 roky
Glenn Jocher
d68179953d
Update export.py
před 4 roky
Naman Gupta
6f3db5e662
Remove autoanchor and class checks on resumed training ( #889 )
* Class frequency not calculated on resuming training
Calculation of class frequency is not needed when resuming training.
Anchors can still be recalculated whether resuming or not.
* Check rank for autoanchor
* Update train.py
no autoanchor checks on resume
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
před 4 roky
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
před 4 roky
Glenn Jocher
d3f9bf2bb7
Update datasets.py
před 4 roky
NanoCode012
901243c780
Fix relative path for VOC download ( #880 )
Cleaned up code
před 4 roky
Glenn Jocher
3e04d20c7d
Merge remote-tracking branch 'origin/master'
před 4 roky
Glenn Jocher
f06e2d518c
opt.image_weights bug fix ( #885 )
před 4 roky
NanoCode012
1e15aad6f9
Add InfiniteDataLoader class ( #876 )
* Add InfiniteDataLoader
Only initializes at first epoch. Saves time.
* Moved class to a better location
před 4 roky
Glenn Jocher
69ff781ca5
opt.img_weights bug fix ( #885 )
před 4 roky
Glenn Jocher
987c226849
seaborn plotting correlogram addition
před 4 roky
Glenn Jocher
08e97a2f88
Update hyperparameters to add lrf, anchors
před 4 roky
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.
před 4 roky
Glenn Jocher
0c01afc390
Update Dockerfile to upgrade pip
před 4 roky
Glenn Jocher
62e037419f
Update Dockerfile FROM nvcr.io/nvidia/pytorch:20.08-py3
před 4 roky
Glenn Jocher
31d6c389f2
Update general.py plot_evolution()
před 4 roky
Glenn Jocher
1e7674d4dd
Update export.py --img-size comment
před 4 roky
Glenn Jocher
a21bd0687c
Update train.py forward simplification
před 4 roky
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`
před 4 roky
Glenn Jocher
5a7d79fbe6
Update general.py bbox_iou() with eps improvements ( #736 )
eps reduced to 1e-9 for additional precaution.
před 4 roky
Glenn Jocher
82ed33a052
generalize webcam detection with opt.source.isnumeric() (yolov3 #1462 )
před 4 roky
Glenn Jocher
09402a2174
torch.from_tensor() bug fix
před 4 roky
Glenn Jocher
4fb8cb353f
Update export.py with v3.0 Hardswish() support ( #831 )
před 4 roky
Glenn Jocher
4d7f222f73
Update export.py with v3.0 Hardswish() support
před 4 roky
Glenn Jocher
8666bc507e
Create evolve.sh for infinite hyp evolution
před 4 roky
Glenn Jocher
a8751e50de
model.fuse() fix for export.py ( #827 )
před 4 roky
Glenn Jocher
5f07782f56
Update requirements.txt for coremltools kmeans quantization
před 4 roky
Glenn Jocher
71209a6099
exportable Hardswish() implementation
před 4 roky
Glenn Jocher
fd71fe8451
Update hyp.finetune.yaml
před 4 roky
Glenn Jocher
f7e075fe93
gsutil cp yaml when evolving to GCP bucket
před 4 roky
Glenn Jocher
9dbaa1bbbe
Update plot_results()
před 4 roky
Glenn Jocher
83dc540b1d
remove ema.ema hasattr(ema, 'module') check
před 4 roky
Glenn Jocher
1309400e6b
Update hyp.finetune.yaml on VOC results
před 4 roky
Glenn Jocher
fb05681a04
Update requirements.txt to coremltools==4.0b3
před 4 roky
Glenn Jocher
d0f98c0095
Update yolo.py with model_info(verbose=False)
před 4 roky
Glenn Jocher
b9851222bf
Merge remote-tracking branch 'origin/master'
před 4 roky
Glenn Jocher
c171e45cda
Update test.py with set() ( #806 )
před 4 roky
Glenn Jocher
ea5e97fb9b
FP16 NMS ( #814 )
před 4 roky
Glenn Jocher
b42e8a531b
optimize imports
před 4 roky
Glenn Jocher
4447f4b937
--resume to same runs/exp directory ( #765 )
* initial commit
* add weight backup dir on resume
před 4 roky
Glenn Jocher
5e0b90de8f
CIoU nan bug fix ( #736 )
před 4 roky
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>
před 4 roky
Glenn Jocher
da35727c68
--save-txt bug fix ( #763 )
před 4 roky
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()
před 4 roky
Francisco Ingham
26c3b11fdc
Update torch_utils.py ( #751 )
před 4 roky
Glenn Jocher
2077d78880
update Dockerfile with libgl1-mesa-glx
před 4 roky