Browse Source

add configuration file and test example

pull/1/MERGE
Marcus Universe 7 years ago
parent
commit
2f6a4cca9b
4 changed files with 79 additions and 0 deletions
  1. +15
    -0
      config.json
  2. +6
    -0
      hyperlpr/config.py
  3. +0
    -0
      hyperlpr/plateStructure.py
  4. +58
    -0
      hyperlpr_test/detect_test.py

+ 15
- 0
config.json View File

@@ -0,0 +1,15 @@
{
"global":{
"debug":true
},
"log":{
},
"detect":{
},
"recognize":{
},
"detectTest":{
"detectPath":"/Users/universe/Documents/work/data/Plate/boundingbox",
"outputPath":"/Users/universe/ProgramUniverse/python/Forked/HyperLPR/out/detect_test"
}
}

+ 6
- 0
hyperlpr/config.py View File

@@ -0,0 +1,6 @@
import json



with open("/Users/universe/ProgramUniverse/python/Forked/HyperLPR/config.json") as f:
configuration = json.load(f)

+ 0
- 0
hyperlpr/plateStructure.py View File


+ 58
- 0
hyperlpr_test/detect_test.py View File

@@ -0,0 +1,58 @@
import os
import hyperlpr.detect as hyperDetect
import hyperlpr.config as hyperConfig
import cv2
import argparse
import sys

parser = argparse.ArgumentParser()

debugInfo = hyperConfig.configuration["global"]["debug"]
testPath = hyperConfig.configuration["detectTest"]["detectPath"]
outPath = hyperConfig.configuration["detectTest"]["outputPath"]

def drawBoundingBox(originImage,rect):
cv2.rectangle(originImage, (int(rect[0]), int(rect[1])), (int(rect[0] + rect[2]), int(rect[1] + rect[3])), (0, 0, 255), 2,
cv2.LINE_AA)
return originImage


#detect Plate in image batch
def detectPlateBatchTest(filepath):
for filename in os.listdir(filepath):
if filename.endswith(".jpg") or filename.endswith(".png"):
fileFullPath = os.path.join(filepath,filename)
image = cv2.imread(fileFullPath)
image_c = image.copy()
Plates = hyperDetect.detectPlateRough(image_c, image_c.shape[0], top_bottom_padding_rate=0.1)
pathName = filename.split('.')[0]
if debugInfo:
if len(Plates) != 0:
if os.path.exists(os.path.join(outPath,pathName)) == False:
os.mkdir(os.path.join(outPath,pathName))
for i,plate in enumerate(Plates):
rect = plate[1]
region = plate[2]
if debugInfo:
cv2.imwrite(os.path.join(outPath,pathName,"region_"+str(i)+"_"+pathName+".png"),region)
drawBoundingBox(image_c,rect)
if debugInfo:
cv2.imwrite(os.path.join(outPath,pathName,"out_"+pathName+".png"),image_c)


def main(args):
if args.type == 'batch':
detectPlateBatchTest(testPath)
else:
print "type: "+args.type+" not found!\n"
print parser.print_help()


def parse_arguments(argv):
parser.add_argument('--type',type=str,help='detect Plate type{batch},default is batch',default='batch')
return parser.parse_args(argv)

if __name__ == "__main__":
main(parse_arguments(sys.argv[1:]))

#detectPlateBatchTest(testPath)

Loading…
Cancel
Save