版权声明:添加我的微信wlagooble,开启一段不一样的旅程 https://blog.csdn.net/nineship/article/details/84138671
from Tkinter import *
import cv2
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import os
import shutil
import csv
file_name_list = []
csv_list = []
now_id = 0
dic_name = ""
csv_name_list = []
csv_value_list = []
csv_save_list = []
def is_normal():
global now_id
if now_id < 0:
now_id = 0
if now_id >= len(file_name_list):
now_id = len(file_name_list)-1
def init():
global dic_name
dic_name = raw_input("dict name: ")
plt.ion()
for root, dirs, files in os.walk("./clip/"+dic_name+"/", topdown=False):
for name in files:
print (name)
file_name_list.append(name)
read_csv("./csv/box_key.csv")
def up_pic():
global now_id
is_normal()
name = "clip/"+dic_name+"/"+file_name_list[now_id]
print (name)
I = mpimg.imread(name)
plt.imshow(I)
plt.show()
now_id -= 1
def next_pic():
global now_id
is_normal()
name = "clip/"+dic_name+"/"+file_name_list[now_id]
print (name)
I = mpimg.imread(name)
plt.imshow(I)
plt.show()
now_id += 1
def move_error():
shutil.move("./clip/"+dic_name+"/"+file_name_list[now_id],"./error/"+dic_name)
file_name_list.pop(now_id)
next_pic()
def del_csv(csv_reader,line):
csv_reader.drop(line)
#csv_reader.close()
return True
def find_csv(jpg_name):
return index
csv_reader = csv.reader(open('./csv/box_key.csv','r'))
for row in csv_reader:
csv_list.append(row)
csv_name_list.append(row[0])
csv_value_list.append(row[1:])
#print ("csv_name_list:",csv_name_list)
#input()
def find_error():
init()
root = Tk()
Button(root, text='up', width=20,height=10,command=up_pic).pack()
Button(root, text='next', width=20,height=10,command=next_pic).pack()
Button(root, text='move', width=20,height=10,command=move_error).pack()
root.mainloop()
def find_list(value):
index = -1
for i in range(len(csv_name_list)):
if csv_name_list[i].find(value) != -1:
index = i
break
return index
#print (csv_name_list)
#print (csv_value_list)
for root, dirs, files in os.walk("./error/", topdown=False):
for name in files:
#print (name)
index = find_list(name)
if index != -1:
continue
print ("index:",index)
#print ("csv_list:",csv_list[index])
#del_csv(csv_reader,index)
#csv_list.pop(index)
csv_save_list.append(csv_list[index])
#input()
csvfile = open("./csv/csv_save/csv_save.csv", 'w')
csvwrite = csv.writer(csvfile)
csvwrite.writerow(csv_save_list)
csvfile.close()
print ("success")