Update to work for python3

Author: imistyrain

demo.py

@@ -4,19 +4,20 @@
 plt.rcParams['image.interpolation'] = 'nearest'
 plt.rcParams['image.cmap'] = 'gray'
 
-import os
-curdir=os.getcwd()
-import sys
-sys.path.insert(0,'../../python')
-
+import os,platform,sys
+if platform.system()=="Windows":
+    caffe_root="D:/CNN/ssd"#2
+else:
+    caffe_root="/home/yanyu/Detection/ssd"
+sys.path.insert(0,caffe_root+'/python')
 import caffe
 caffe.set_device(0)
 caffe.set_mode_gpu()
 from google.protobuf import text_format
 from caffe.proto import caffe_pb2
 datasetname="Face2017"
 # load PASCAL VOC labels
-labelmap_file = curdir+'/labelmap_face.prototxt'
+labelmap_file = 'labelmap_face.prototxt'
 file = open(labelmap_file, 'r')
 labelmap = caffe_pb2.LabelMap()
 text_format.Merge(str(file.read()), labelmap)
@@ -28,7 +29,7 @@ def get_labelname(labelmap, labels):
         labels = [labels]
     for label in labels:
         found = False
-        for i in xrange(0, num_labels):
+        for i in range(0, num_labels):
             if label == labelmap.item[i].label:
                 found = True
                 labelnames.append(labelmap.item[i].display_name)
@@ -38,8 +39,8 @@ def get_labelname(labelmap, labels):
 #model_def =curdir+"/cpp/models/face_deploy.prototxt"
 #model_weights=curdir+"/cpp/models/VGG_Face2017_SSD_300x300_iter_120000.caffemodel"
 #image_resize = 300
-model_def =curdir+"/cpp/models/faceboxes_deploy.prototxt"
-model_weights=curdir+"/cpp/models/FaceBoxes_1024x1024.caffemodel"
+model_def ="cpp/models/faceboxes_deploy.prototxt"
+model_weights="cpp/models/FaceBoxes_1024x1024.caffemodel"
 image_resize = 1024
 net = caffe.Net(model_def,      # defines the structure of the model
                 model_weights,  # contains the trained weights
@@ -85,13 +86,13 @@ def get_labelname(labelmap, labels):
 plt.imshow(image)
 currentAxis = plt.gca()
 
-for i in xrange(top_conf.shape[0]):
+for i in range(top_conf.shape[0]):
     xmin = int(round(top_xmin[i] * image.shape[1]))
     ymin = int(round(top_ymin[i] * image.shape[0]))
     xmax = int(round(top_xmax[i] * image.shape[1]))
     ymax = int(round(top_ymax[i] * image.shape[0]))
     score = top_conf[i]
-    print xmin,ymin,xmax,ymax,score
+    print(xmin,ymin,xmax,ymax,score)
     label = int(top_label_indices[i])
     label_name = top_labels[i]
     display_txt = '%s: %.2f'%(label_name, score)

faceboxes_test.py

@@ -1,7 +1,11 @@
 # -*- coding: utf-8 -*
 import numpy as np  
-import sys,os,cv2
-caffe_root = 'D:/CNN/ssd'
+import sys,os,cv2,platform
+if platform.system()=="Windows":
+    caffe_root="D:/CNN/ssd"
+else:
+    caffe_root="/home/yanyu/Detection/ssd"
+sys.path.insert(0,caffe_root+'/python')
 sys.path.insert(0, caffe_root + '/python')  
 import caffe  
 import time
@@ -48,7 +52,7 @@ def testimg(imgfile):
     time_start=time.time()
     out = net.forward()  
     time_end=time.time()
-    print time_end-time_start
+    print(time_end-time_start)
     box, conf, cls = postprocess(frame, out)
     for i in range(len(box)):
         p1 = (box[i][0], box[i][1])
@@ -79,8 +83,7 @@ def detect(cameraindex=0):
       time_start=time.time()
       out = net.forward()  
       time_end=time.time()
-      print time_end-time_start,  
-      print "s"
+      print(time_end-time_start)
 
       box, conf, cls = postprocess(frame, out)
 

ssd-face.pyproj

@@ -5,7 +5,7 @@
     <SchemaVersion>2.0</SchemaVersion>
     <ProjectGuid>69443bc6-e3e9-42ba-a8d2-9c8075b1a700</ProjectGuid>
     <ProjectHome>.</ProjectHome>
-    <StartupFile>create_all.py</StartupFile>
+    <StartupFile>faceboxes_test.py</StartupFile>
     <SearchPath>
     </SearchPath>
     <WorkingDirectory>.</WorkingDirectory>
@@ -30,6 +30,7 @@
   <ItemGroup>
     <Compile Include="create_all.py" />
     <Compile Include="demo.py" />
+    <Compile Include="faceboxes_test.py" />
     <Compile Include="test.py" />
     <Compile Include="train.py" />
   </ItemGroup>

test.py

@@ -1,6 +1,10 @@
 from __future__ import print_function
-import sys
-sys.path.append('python')
+import sys,platform
+if platform.system()=="Windows":
+    caffe_root="D:/CNN/ssd"
+else:
+    caffe_root="/home/yanyu/Detection/ssd"
+sys.path.insert(0,caffe_root+'/python')
 import caffe
 from caffe.model_libs import *
 from google.protobuf import text_format

train.py

@@ -1,7 +1,7 @@
 from __future__ import print_function
 import sys,platform
 if platform.system()=="Windows":
-    caffe_root="D:/CNN/ssd"#2
+    caffe_root="D:/CNN/ssd"
 else:
     caffe_root="/home/yanyu/Detection/ssd"
 sys.path.insert(0,caffe_root+'/python')