Merge pull request BVLC#161 from gheinrich/dev/explicit-class-mapping

Explicit class mapping in DetectNet
lukeyeager · Jun 8, 2016 · 846f3e4 · 846f3e4
2 parents 883597f + 16abdb2
commit 846f3e4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/kitti/detectnet_network.prototxt b/examples/kitti/detectnet_network.prototxt
@@ -79,6 +79,7 @@ layer {
     image_size_x: 1248
     image_size_y: 384
     crop_bboxes: true
+    object_class: { src: 1 dst: 0} # obj class 1 -> cvg index 0
   }
    detectnet_augmentation_param: {
     crop_prob: 1
@@ -117,6 +118,7 @@ layer {
     image_size_x: 1248
     image_size_y: 384
     crop_bboxes: false
+    object_class: { src: 1 dst: 0} # obj class 1 -> cvg index 0
   }
   transform_param: {
     mean_value: 127