[darknet]查看错误结果 sight of wrong

佚名 5年前 (2019-02-28) 人工智能 1643人围观抢沙发百度已收录

import os
import numpy
import cv2

bad_label_file = open("bad_valid.list",'r')

names = []
for line in open('data/coco.names','r'):
    names.append(line.strip('\n'))
classes = len(names)

colors = []
for i in range(80):
    #color = [random.uniform(0, 255),random.uniform(0, 255),random.uniform(0, 255)]
    color = [(i)*255/classes,(classes-i)*255/classes,(classes+i)*128/classes]
    colors.append(color)


lines = []
for line in bad_label_file:
    lines.append(line)

mount = len(lines)

oper=0
i=0

#for i in range(mount):
while(i<mount):
    i+=oper
    
    words = lines[i].split(':')
    imagepath = words[0]
    #print('reading "{}: No. {}"'.format(path,indexes))
    print('{}\\{}'.format(mount,i))
    
    src = cv2.imread(imagepath)
    #cv2.imshow("src",src)
    h,w = src.shape[:2]


    labelpath = imagepath.split('-')[0]+'-labels/'+os.path.basename(imagepath).split('.')[0]+'.txt'
    try:
        labelfile = open(labelpath,'r')
        labels = []
        for line in labelfile:
            #print('get label {}'.format(line))
            labels.append(line)
    
        for label in labels:
            ti,tx,ty,tw,th = map(float,label.split(' ')[:5])
            #print('get label {} {} {} {} {}'.format(ti,tx,ty,tw,th))
            x0=w*(tx-tw/2)
            y0=h*(ty-th/2)
            x1=w*(tx+tw/2)
            y1=h*(ty+th/2)
            w0=w*tw
            h0=h*th
            ti,x0,y0,x1,y1,w0,h0 = map(int,[ti,x0,y0,x1,y1,w0,h0])
            #print('<({},{}),({},{})>,'.format(x0,y0,x1,y1))
            cv2.rectangle(src,(x0,y0),(x1,y1),colors[ti],2)
            y0+=15
            #cv2.putText(src,str(ti)+' '+names[ti],(x0+1,y0+1),cv2.FONT_HERSHEY_COMPLEX,0.5,colors[ti][::-1],1)
            #cv2.putText(src,str(ti)+' '+names[ti],(x0-1,y0-1),cv2.FONT_HERSHEY_COMPLEX,0.5,colors[ti][::-1],1)
            cv2.putText(src,str(ti)+' '+names[ti],(x0,y0),cv2.FONT_HERSHEY_COMPLEX,0.5,colors[ti],1)
    except :
        print('guess what? ERROR!(show_bad_valid.py::67)')

    
    words = words[1].strip('\n')
    probs = words.split(' ')[2:]
    #print(words)
    #print(probs)
    if len(probs) > 0:
        for prob in probs:
            values = prob.strip('<').strip('>').split(',')
            print(values)
            ti,tx,ty,tw,th = map(float,values[2:])
            x0=w*(tx-tw/2)
            y0=h*(ty-th/2)
            x1=w*(tx+tw/2)
            y1=h*(ty+th/2)
            ti,x0,y0,x1,y1 = map(int,[ti,x0,y0,x1,y1])
            y1-=10
            cv2.rectangle(src,(x0,y0),(x1,y1),colors[ti][::-1],1)
            cv2.putText(src,str(ti)+' '+names[ti],(x0,y1),cv2.FONT_HERSHEY_COMPLEX,0.5,colors[ti],1)


    cv2.imshow("image",src)

    key = cv2.waitKey(0)
    '''
    A 65
    a 97
    D 68
    d 100
    S 83
    s 115
    '''
    if key == 97:
        if i>0:
            oper=-1
    elif key == 100:
        if i<len(lines):
            oper=1
    elif key == 115:
        oper=0
        cv2.imwrite('workingspace.jpg',src)
    else:
        break
bad_label_file.close()