adding yolo

.gitignore

@@ -1,3 +1,4 @@
+*.weights
 *.pkl
 *.pyc
 *.swp

Run.py

@@ -28,6 +28,7 @@
 from library.Math import *
 from library.Plotting import *
 from torch_lib import Model, ClassAverages
+from yolo.yolo import cv_Yolo
 
 import os
 import cv2
@@ -72,6 +73,11 @@ def main():
     dataset = Dataset(os.path.abspath(os.path.dirname(__file__)) + '/eval')
     averages = ClassAverages.ClassAverages()
 
+    yolo_path = os.path.abspath(os.path.dirname(__file__)) + '/weights'
+    yolo = cv_Yolo(yolo_path)
+
+    exit()
+
     all_images = dataset.all_objects()
     for key in sorted(all_images.keys()):
         data = all_images[key]

weights/coco.names

@@ -0,0 +1,80 @@
+person
+bicycle
+car
+motorbike
+aeroplane
+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
+sofa
+pottedplant
+bed
+diningtable
+toilet
+tvmonitor
+laptop
+mouse
+remote
+keyboard
+cell phone
+microwave
+oven
+toaster
+sink
+refrigerator
+book
+clock
+vase
+scissors
+teddy bear
+hair drier
+toothbrush

weights/get_weights.sh

@@ -1,2 +1,5 @@
 #! /bin/bash
+echo "downloading 3d bbox weights ..."
 wget --load-cookies /tmp/cookies.txt "https://docs.google.com/uc?export=download&confirm=$(wget --quiet --save-cookies /tmp/cookies.txt --keep-session-cookies --no-check-certificate 'https://docs.google.com/uc?export=download&id=1yEiquJg9inIFgR3F-N5Z3DbFnXJ0aXmA' -O- | sed -rn 's/.*confirm=([0-9A-Za-z_]+).*/\1\n/p')&id=1yEiquJg9inIFgR3F-N5Z3DbFnXJ0aXmA" -O epoch_10.pkl && rm -rf /tmp/cookies.txt
+echo "downloading yolo weights ..."
+wget https://pjreddie.com/media/files/yolov3.weights