9f04e175f6
nms torch.mm() update
2020-05-10 11:26:37 -07:00
ae2bc020eb
git status check - linux and darwin
2020-05-09 22:35:44 -07:00
965155ee60
CUBLAS bug fix #1139
2020-05-06 10:26:28 -07:00
ee7cba65a5
kmeans() cleanup
2020-05-02 09:20:19 -07:00
b0629d622c
bug fix on #1114
2020-04-30 15:03:32 -07:00
0ffbf5534e
cleanup for #1114
2020-04-30 14:53:57 -07:00
fb1b5e09b2
faster and more informative training plots ( #1114 )
...
* faster and more informative training plots
* Update utils.py
Looks good. Needs pep8 linting, I'll do that in PyCharm later once PR is in.
* Update test.py
* Update train.py
f for the tb descriptor lets us plot several batches, i.e. to allow us to change L292 to 'if ni < 3' for 3 examples.
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-04-30 13:37:04 -07:00
754a1b5bf8
reduce merge limit to 3000
2020-04-25 21:15:30 -07:00
77b3829d56
check_git_status() to train.py
2020-04-22 11:02:09 -07:00
accce6b565
git status check bug fix
2020-04-18 18:06:11 -07:00
510eadcfa5
Apex and 'git pull' suggestions
2020-04-15 12:54:56 -07:00
ac4c90c817
cleanup
2020-04-14 13:08:00 -07:00
763cdd5ae2
detailed image sizes report
2020-04-14 11:51:19 -07:00
029e137bc2
bug fix
2020-04-14 04:34:40 -07:00
1681249588
cleanup
2020-04-14 04:15:53 -07:00
198a5a591d
code cleanup
2020-04-14 04:15:05 -07:00
25725c8569
bug fix
2020-04-14 03:13:30 -07:00
76fb8d48d4
ng dependence removed from build_targets()
2020-04-13 21:25:03 -07:00
efc754a794
add generations arg to kmeans()
2020-04-12 12:49:23 -07:00
b574f765ce
add warning to plot_results()
2020-04-11 11:04:10 -07:00
58edfc4a84
kaiming weight init
2020-04-11 10:45:33 -07:00
398f8eadec
add thr=0.10 to kmean_anchors()
2020-04-10 16:34:32 -07:00
4120ac3aa6
training updates
2020-04-08 21:01:58 -07:00
a19b1a3b94
line thickness
2020-04-05 11:10:05 -07:00
6203340888
detect.py multi_label default False
2020-04-05 11:05:49 -07:00
207c6fcff9
merge NMS full matrix
2020-04-02 18:53:40 -07:00
300e9a7ad6
merge NMS full matrix
2020-03-31 21:31:09 -07:00
02802e67f2
merge NMS full matrix
2020-03-31 18:18:08 -07:00
f4eecef700
merge NMS speed/memory improvements
2020-03-31 15:37:23 -07:00
eb151a881e
NMS and test batch_size updates
2020-03-29 20:41:32 -07:00
f9d34587da
Merge NMS update
2020-03-27 13:11:24 -07:00
582de735ad
utils.clip_coords doesn't work as expected. ( #961 )
...
* utils.clip_coords doesn't work as expected.
Box coords may be negative or exceed borders.
* Update utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-03-27 13:09:10 -07:00
4a63b24b09
Merge NMS update
2020-03-26 19:50:29 -07:00
01ee0c5e95
Merge NMS update
2020-03-26 18:41:04 -07:00
dad59220f1
speed and comments update
2020-03-26 18:34:20 -07:00
a322fc5d4b
Merge NMS update
2020-03-26 12:48:00 -07:00
171b4129b5
Merge NMS update
2020-03-26 12:33:12 -07:00
eac07f9da3
Merge NMS update
2020-03-26 12:20:01 -07:00
23b34f4db8
merge_batch NMS method
2020-03-25 23:29:33 -07:00
aa0c64b5ac
merge_batch NMS method
2020-03-25 23:24:57 -07:00
60c8d194cd
FocalLoss() and obj loss speed and stability update
2020-03-19 11:52:52 -07:00
b3adc896f9
focal and obj loss speed/stability
2020-03-16 21:40:57 -07:00
448c4a6e1f
Remove deprecated --arc architecture options, implement --arc default for all cases
2020-03-16 20:46:25 -07:00
1a12667ce1
loss function cleanup
2020-03-16 17:31:37 -07:00
c09fcfc4fe
EMA class updates
2020-03-16 14:18:56 -07:00
c4047000fe
FocalLoss() updated to match TF
2020-03-16 14:03:50 -07:00
666ba85ed3
Comment updates on box coordinates ( #852 )
...
* Update utils.py
Reusing function defined above
* Update utils.py
* Reverting change which break bbox coordinate computation
* Update utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-03-14 15:35:59 -07:00
418269d739
FocalLoss() gamma and alpha default values
2020-03-13 16:51:30 -07:00
f30a8706e5
update to coco results82
2020-03-13 12:06:48 -07:00
0de07da612
updates
2020-03-13 11:03:39 -07:00
Glenn Jocher
Josh Veitch-Michaelis
GoogleWiki
Falak