Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

YOLO loss #2376

Merged
merged 65 commits into from
Jul 30, 2021
Merged

YOLO loss #2376

Show file tree
Hide file tree
Changes from 2 commits
Commits
Show all changes
65 commits
Select commit Hold shift + click to select a range
a49a42d
Initial version of the loss_yolo layer, still not working
arrufat Jun 6, 2021
dccbaca
add option to visualize the detections of the last training synchroni…
arrufat Jun 7, 2021
66cd401
wip: add more data augmentation, remove tag from input and use the sa…
arrufat Jun 8, 2021
337b122
wip: use a custom yolo_rect with drectangle instead of rectangle
arrufat Jun 11, 2021
9324c66
wip: rename to tensor_to_loss, builtin iou and same scale for grad an…
arrufat Jun 13, 2021
e557be3
wip: increment truth iterator after each sample
arrufat Jun 22, 2021
5842b0d
use the L2 loss for display purposes and serialize labels
arrufat Jun 23, 2021
e82b8f5
update training code
arrufat Jun 23, 2021
729223a
use options.overlaps_ignore field
arrufat Jun 23, 2021
4a129f5
make random_cropper work with yolo_rect types
arrufat Jun 23, 2021
dfd56c1
add some comments to the example training program
arrufat Jun 23, 2021
9ea1a02
do not use overlaps_ignore
arrufat Jun 23, 2021
fe2d7e5
add mAP computation
arrufat Jun 25, 2021
c35db67
print progress while computing map
arrufat Jun 25, 2021
df8d64c
use explicit type
arrufat Jun 25, 2021
9d703a2
fix mAP computation and initialize color mapper for all labels
arrufat Jun 26, 2021
05ac6ff
allow multi label detections
arrufat Jun 26, 2021
7f9c716
simplify multilabel processing output
arrufat Jun 26, 2021
c07910c
add documentation
arrufat Jun 26, 2021
f6cae91
remove test image added by mistake
arrufat Jun 26, 2021
adfbabc
fix documentation, loss details printing and compilation on windows
arrufat Jun 27, 2021
23c021e
add optional height and width fields to image_dataset_metadata::image
arrufat Jun 27, 2021
1211bc3
add simple anchor boxes computation using K-Means clustering
arrufat Jun 27, 2021
2499947
fix comment
arrufat Jun 27, 2021
a2959ce
remove unused variable
arrufat Jun 27, 2021
7474ccc
fix maybe uninitialized value
arrufat Jun 27, 2021
c96b9de
fix sign compare warning
arrufat Jun 27, 2021
9becb23
print iou_ignore_threshold
arrufat Jun 28, 2021
c63738a
remove unused field
arrufat Jun 28, 2021
9a0f0a3
improve loss formatting
arrufat Jun 28, 2021
1fbc41b
fix tag output formatting
arrufat Jun 28, 2021
3fba65f
fix documentation and add classwise to adjust_nms
arrufat Jul 2, 2021
0115778
minor formatting fix in loss
arrufat Jul 2, 2021
9723dd4
add option to allow objects to fall outside of the crops output by th…
arrufat Jul 3, 2021
3c0e6c2
add iou_anchor_threshold
arrufat Jul 6, 2021
8a7ebb6
simplify anchor update logic
arrufat Jul 6, 2021
45498c8
oops, fix typo
arrufat Jul 6, 2021
3d2ae14
regression fix: do not update other anchors with low IoU
arrufat Jul 6, 2021
32021ae
make NMS classwise by default
arrufat Jul 6, 2021
00c3c43
Make imglab populate the new width and height image fields
davisking Jul 10, 2021
040c83f
update example xml files to have width and height annotations
davisking Jul 10, 2021
0eeb9de
Adjust example so it builds in C++11 and slightly simplified code.
davisking Jul 10, 2021
f0fff72
Add --add-width-height-metadata to imglab
davisking Jul 10, 2021
4f80d8d
remove redundant check
arrufat Jul 12, 2021
c3b5fde
Merge branch 'loss-yolo' of https://github.com/arrufat/dlib into loss…
arrufat Jul 12, 2021
6e51086
update comment about iou_anchor_threshold
arrufat Jul 13, 2021
ffec8e7
clarify more comment about iou_anchor_threshold
arrufat Jul 13, 2021
1e93f86
Merge branch 'davisking:master' into loss-yolo
arrufat Jul 27, 2021
74af452
use long, images have nr and nc defined as long (avoids warnings)
arrufat Jul 27, 2021
f641c21
Merge branch 'loss-yolo' of https://github.com/arrufat/dlib into loss…
arrufat Jul 27, 2021
f888553
Update dlib/dnn/loss_abstract.h
arrufat Jul 29, 2021
000ffae
Update dlib/dnn/loss_abstract.h
arrufat Jul 29, 2021
9d5c032
Update dlib/dnn/loss_abstract.h
arrufat Jul 29, 2021
8822c1b
Update dlib/dnn/loss_abstract.h
arrufat Jul 29, 2021
c747def
Update dlib/image_processing/full_object_detection_abstract.h
arrufat Jul 29, 2021
66fe53e
Update dlib/image_transforms/random_cropper_abstract.h
arrufat Jul 29, 2021
281b941
Update examples/dnn_yolo_train_ex.cpp
arrufat Jul 29, 2021
a6954ee
Update examples/dnn_yolo_train_ex.cpp
arrufat Jul 29, 2021
5efdb0c
Update dlib/dnn/loss_abstract.h
arrufat Jul 29, 2021
c671f05
update random_cropper docs about get_min_object_coverage
arrufat Jul 29, 2021
b5815b3
improve classwise_nms documentation
arrufat Jul 29, 2021
1902a60
fix adjust_nms documentation
arrufat Jul 29, 2021
7c0b425
Revert "fix adjust_nms documentation"
arrufat Jul 29, 2021
fdd49cf
split line
arrufat Jul 29, 2021
4793705
fix description about the returned loss value
arrufat Jul 30, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading