adds_yolo_model

src/confs/config_coco.json

@@ -0,0 +1,33 @@
+{
+    "model" : {
+        "input_size":       416,
+        "grid_size":        13,
+        "true_box_buffer":  10,
+        "iou_threshold":    0.5,  
+        "nms_threshold":    0.3
+    },
+    "config_path" : {
+        "labels":           "models/coco/labels_coco.txt",
+        "anchors":          "models/coco/anchors_coco.txt",
+        "arch_plotname":    ""
+    },
+    "train": {
+        "out_model_name":   "",
+        "image_folder":     "",
+        "annot_folder":     "",
+        "batch_size":       16,
+        "learning_rate":    1e-4,
+        "num_epochs":       20,
+        "object_scale":     5.0 ,
+        "no_object_scale":  1.0,
+        "coord_scale":      1.0,
+        "class_scale":      1.0,
+        "verbose":          1
+    },
+
+    "valid": {
+        "image_folder":     "",
+        "annot_folder":     "",
+        "pred_folder":      ""
+    }
+}

src/confs/config_voc.json

@@ -0,0 +1,34 @@
+{
+    "model" : {
+        "input_size":       416,
+        "grid_size":        13,
+        "true_box_buffer":  10,
+        "iou_threshold":    0.5,  
+        "nms_threshold":    0.45
+    },
+    "config_path" : {
+        "labels":           "models/voc/labels_voc.txt",
+        "anchors":          "models/voc/anchors_voc.txt",
+        "arch_plotname":    "voc_arch.png"
+    },
+    "train": {
+        "out_model_name":   "yolo_retrained_voc.h5",
+        "image_folder":     "/home/kiran/Documents/DATA/VOC/train/imgs",
+        "annot_folder":     "/home/kiran/Documents/DATA/VOC/train/anns",
+        "batch_size":       16,
+        "learning_rate":    1e-4,
+        "num_epochs":       50,
+        "object_scale":     5.0 ,
+        "no_object_scale":  1.0,
+        "coord_scale":      1.0,
+        "class_scale":      1.0,
+        "verbose":          1
+    },
+
+    "valid": {
+        "image_folder":     "/home/kiran/Documents/DATA/VOC/valid/imgs",
+        "annot_folder":     "/home/kiran/Documents/DATA/VOC/valid/anns",
+        "pred_folder":      "/home/kiran/Documents/DATA/VOC/valid/img_pred",
+        "plot_preds":       true
+    }
+}

src/gpu_test.py

@@ -0,0 +1,2 @@
+from keras import backend as K
+K.tensorflow_backend._get_available_gpus()

src/models/coco/anchors_coco.txt

@@ -0,0 +1 @@
+0.57273, 0.677385, 1.87446, 2.06253, 3.33843, 5.47434, 7.88282, 3.52778, 9.77052, 9.16828

src/models/coco/labels_coco.txt

@@ -0,0 +1,80 @@
+person
+bicycle
+car
+motorcycle
+airplane
+bus
+train
+truck
+boat
+traffic light
+fire hydrant
+stop sign
+parking meter
+bench
+bird
+cat
+dog
+horse
+sheep
+cow
+elephant
+bear
+zebra
+giraffe
+backpack
+umbrella
+handbag
+tie
+suitcase
+frisbee
+skis
+snowboard
+sports ball
+kite
+baseball bat
+baseball glove
+skateboard
+surfboard
+tennis racket
+bottle
+wine glass
+cup
+fork
+knife
+spoon
+bowl
+banana
+apple
+sandwich
+orange
+broccoli
+carrot
+hot dog
+pizza
+donut
+cake
+chair
+couch
+potted plant
+bed
+dining table
+toilet
+tv
+laptop
+mouse
+remote
+keyboard
+cell phone
+microwave
+oven
+toaster
+sink
+refrigerator
+book
+clock
+vase
+scissors
+teddy bear
+hair drier
+toothbrush

src/models/voc/anchors_voc.txt

@@ -0,0 +1 @@
+1.3221, 1.73145, 3.19275, 4.00944, 5.05587, 8.09892, 9.47112, 4.84053, 11.2364, 10.0071

src/models/voc/labels_voc.txt

@@ -0,0 +1,20 @@
+aeroplane
+bicycle
+bird
+boat
+bottle
+bus
+car
+cat
+chair
+cow
+diningtable
+dog
+horse
+motorbike
+person
+pottedplant
+sheep
+sofa
+train
+tvmonitor

src/net/__init__.py

@@ -0,0 +1,7 @@
+#from . import netarch
+#from . import netdecode
+#from . import neteval
+#from . import netgen
+#from . import netloss
+#from . import netparams
+#from . import utils