def __init__(self,condition='light'): self.condition=condition self.picture_path='./pictures/'+condition+'/' self.label_name='./labels/'+condition+'_label.npy' self.feature_file='./labels/'+condition+'_labels.csv' # self.counter=0 self.img_reader=ImgReader(self.picture_path,self.label_name) self.cube=CubeBase() self.color_reader=ColorReader(self.feature_file) self.verifier=ValidateClass()
class Controller(): def __init__(self, condition='light'): self.condition = condition self.picture_path = './pictures/' + condition + '/' self.label_name = './labels/' + condition + '_label.npy' self.feature_file = './labels/' + condition + '_labels.csv' self.center_file = './labels/' + condition + '_center.npy' # self.counter=0 self.img_reader = ImgReader(self.picture_path, self.label_name) self.center = CenterReader(self.center_file) self.cube = CubeBase() self.color_reader = ColorReader(self.feature_file) self.verifier = ValidateClass() def run(self, maxgroup=11): for i in range(maxgroup): # print("group:",i) img = self.img_reader.update(i) center = self.center.getCenter(i) # print(center) # result=self.cube.onlineUpdate(img,center) result = self.cube.test(img) self.color_reader.update(i) self.verifier.validate(i, result, self.color_reader.cur_labels) print("dataset:", self.condition) self.verifier.showResult(isALL=False)
import numpy as np import cv2 from Classes import ImgReader, CubeBase, ValidateClass, ColorReader condition = 'light' picture_path = './pictures/' + condition + '/' label_name = './labels/' + condition + '_label.npy' feature_file = './labels/' + condition + '_feature.csv' # cv2.namedWindow("img") counter = 0 #img_reader img_reader = ImgReader(picture_path, label_name) img = img_reader.update(counter) #cube cube = CubeBase() result = cube.test(img) result_img = cube.drawResult() #color_reader color_reader = ColorReader(feature_file) color_img = color_reader.update() #Verifier verifier = ValidateClass() verifier.validate(counter, result, color_reader.labels) verifier.showResult() counter += 1 cv2.imshow('img', img) cv2.imshow('result', result_img) while True: key = cv2.waitKey(1)
from Classes import ImgReader, CubeBase, ColorReader # conditions=["cube1_dark","cube1_light","cube2_light","cube2_right_light","cube2_up_light"] conditions = ["cube2_light"] feature = "hsv" for condition in conditions: picture_path = './pictures/' + condition + '/' label_name = './labels/' + condition + '_label.npy' feature_file = './labels/' + condition + '_' + feature + '_feature.csv' roi_path = picture_path + 'rois/' print("The condition is %s" % (condition)) cv2.namedWindow("img") counter = 0 img_reader = ImgReader(picture_path, label_name) cube = CubeBase() img = img_reader.update(counter) color_reader = ColorReader(feature_file) color_img = color_reader.update(counter) while True: # hist=calSoftHist(img_reader.colors[count]) # drawHist(hist) cv2.imshow('img', img) cv2.imshow('color_img', color_img) key = cv2.waitKey(1) if key == 27: break if (key == ord('n')): try: counter += 1
import numpy as np import cv2 from Classes import ImgReader, CubeBase, ValidateClass, ColorReader condition = 'light' picture_path = './pictures/' + condition + '/' label_name = './labels/' + condition + '_label.npy' feature_file = './labels/' + condition + '_feature.csv' cv2.namedWindow("img") counter = 0 #img_reader img_reader = ImgReader(picture_path, label_name) img = img_reader.update(counter) #cube cube = CubeBase() result = cube.update(img) result_img = cube.drawResult() #color_reader color_reader = ColorReader(feature_file) color_img = color_reader.update() #Verifier verifier = ValidateClass() verifier.validate(counter, result, color_reader.labels) verifier.showResult() counter += 1 cv2.imshow('img', img) cv2.imshow('result', result_img) while True: key = cv2.waitKey(1)
from Classes import ImgReader, CubeBase conditions = [ "cube1_dark", "cube1_light", "cube2_light", "cube2_right_light", "cube2_up_light" ] for condition in conditions: features = "all" picture_path = './pictures/' + condition + '/' label_name = './labels/' + condition + '_label.npy' # feature_file='./labels/'+condition+'_'+features+'_feature.csv' feature_file = './labels/' + "feature_" + condition + '.csv' roi_path = picture_path + 'rois/' cv2.namedWindow("img") counter = 0 img_reader = ImgReader(picture_path, label_name) cube = CubeBase() img = img_reader.update(counter) cube.saveFeatures(img, feature_file, counter, features) cv2.imwrite(roi_path + str(counter) + '.jpg', img) counter += 1 try: while True: # hist=calSoftHist(img_reader.colors[count]) # drawHist(hist) cv2.imshow('img', img) key = cv2.waitKey(1) if key == 27: break if (key == ord('n')): print('save %d features' % (counter)) img = img_reader.update(counter)