Alex Stoken
|
25e51bcec7
|
add util function to get most recent last.pt file
added logic in train.py __main__ to handle resuming from a run
|
5 years ago |
|
Alex Stoken
|
490f1e7b9c
|
add save_dir arg to plot_lr_scheduler, default to current dir.
Uncomment plot_lr_scheduler in train() and pass log_dir as save location
|
5 years ago |
Glenn Jocher
|
8b26e89006
|
AutoAnchor bug fix #72
|
5 years ago |
|
Glenn Jocher
|
8fa3724072
|
kmean_anchors() update
|
5 years ago |
|
Glenn Jocher
|
1c0b6236e3
|
update fast mode
|
5 years ago |
|
Glenn Jocher
|
db2c3acd3a
|
updated testing settings, rebalanced towards FP16 latency
|
5 years ago |
|
Glenn Jocher
|
c5966abba8
|
glob search bug fix #77
|
5 years ago |
|
Glenn Jocher
|
260b1729f0
|
FP16 inference update
|
5 years ago |
|
Glenn Jocher
|
93bbf1b87a
|
best possible recall fix
|
5 years ago |
|
Glenn Jocher
|
31f3310029
|
assert best possible recall > 0.9 before training
|
5 years ago |
|
Glenn Jocher
|
099e6f5ebd
|
--img-size stride-multiple verification
|
5 years ago |
|
Glenn Jocher
|
3a5c5328c5
|
ONNX, BCEBlurWithLogitsLoss, plot_study updates
|
5 years ago |
|
Glenn Jocher
|
c14368d768
|
new CSP model release
|
5 years ago |
|
Glenn Jocher
|
ff02ae0869
|
module updates
|
5 years ago |
|
Glenn Jocher
|
1f698af6f1
|
create_backbone with SourceChangeWarning update
|
5 years ago |
|
Glenn Jocher
|
66d73e4cf3
|
NMS updates
|
5 years ago |
|
Glenn Jocher
|
948bcdd219
|
--classes bug fix #17
|
5 years ago |
|
Glenn Jocher
|
0362f103ef
|
updates
|
5 years ago |
|
Glenn Jocher
|
eb97b2e413
|
NMS fast mode
|
5 years ago |
|
Glenn Jocher
|
b10609fe6b
|
updates
|
5 years ago |
|
Glenn Jocher
|
72eed215ff
|
updates
|
5 years ago |
|
Glenn Jocher
|
5a79b5f65c
|
updates
|
5 years ago |
|
Glenn Jocher
|
4f44aaf26b
|
updates
|
5 years ago |
|
Glenn Jocher
|
1e84a23f38
|
initial commit
|
5 years ago |
