Skip to content

Commit 16abdb2

Browse files
committed
Explicit class mapping in DetectNet
1 parent 057ace8 commit 16abdb2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/kitti/detectnet_network.prototxt

+2
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,7 @@ layer {
7979
image_size_x: 1248
8080
image_size_y: 384
8181
crop_bboxes: true
82+
object_class: { src: 1 dst: 0} # obj class 1 -> cvg index 0
8283
}
8384
detectnet_augmentation_param: {
8485
crop_prob: 1
@@ -117,6 +118,7 @@ layer {
117118
image_size_x: 1248
118119
image_size_y: 384
119120
crop_bboxes: false
121+
object_class: { src: 1 dst: 0} # obj class 1 -> cvg index 0
120122
}
121123
transform_param: {
122124
mean_value: 127

0 commit comments

Comments
 (0)