#fe.loadSuperpixelFromFile(sp_file_names[i]) fe.loadLabelImage(label_file_names[i]) featureVectors = fe.getFeaturesVectors() labels = fe.getSuperPixelLabels() #Test purposes edges, edgeFeatures1, edgeFeatures2 = fe.getEdges() if file_labels[i] != TESTING_LABEL: # store data if file_labels[i] == TRAINING_LABEL: train_edges.append(edges) train_edgesFeatures1.append(edgeFeatures1) train_edgesFeatures2.append(edgeFeatures2) train_superpixels.append(fe.getSuperpixelImage()) train_labels = np.append(train_labels, labels, 0) if train_data == []: train_data = featureVectors else: train_data = np.vstack((train_data, featureVectors)) else: # get superpixel valid files valid_edges.append(edges) valid_edgesFeatures1.append(edgeFeatures1) valid_edgesFeatures2.append(edgeFeatures2) valid_superpixels.append(fe.getSuperpixelImage()) validationOriginalImage.append(im_file_names[i]) # these two lines need to be added into featureExtraction class valid_files = sp.getSuperValidFiles(fe.getSuperpixelImage(), valid_files_count, valid_files)
#fe.loadSuperpixelFromFile(sp_file_names[i]) fe.loadLabelImage(label_file_names[i]) featureVectors = fe.getFeaturesVectors() labels = fe.getSuperPixelLabels() #Test purposes edges, edgeFeatures1, edgeFeatures2 = fe.getEdges() if file_labels[i] != TESTING_LABEL: # store data if file_labels[i] == TRAINING_LABEL: train_edges.append(edges) train_edgesFeatures1.append(edgeFeatures1) train_edgesFeatures2.append(edgeFeatures2) train_superpixels.append(fe.getSuperpixelImage()) train_labels = np.append(train_labels, labels, 0) if train_data==[]: train_data = featureVectors else: train_data = np.vstack((train_data,featureVectors)) else: # get superpixel valid files valid_edges.append(edges) valid_edgesFeatures1.append(edgeFeatures1) valid_edgesFeatures2.append(edgeFeatures2) valid_superpixels.append(fe.getSuperpixelImage()) validationOriginalImage.append(im_file_names[i]) # these two lines need to be added into featureExtraction class valid_files = sp.getSuperValidFiles(fe.getSuperpixelImage(), valid_files_count, valid_files) valid_pixels_labels.append(sp.getPixelLabel(fe.getLabelImage()))