getpatch
import time import os import math import sys import os,os.path,shutil import numpy as np import cv2 img_in_path = 'F:/project/Breast/InBreast/INBreast/removeother/' # img_in_path = 'F:/project/Breast/InBreast/INBreast/imgtest/' outpathimgyes = 'F:/project/Breast/InBreast/INBreast/outimgpatch/calcification/' outpathimgno = 'F:/project/Breast/InBreast/INBreast/outimgpatch/no/' txtPath = 'F:/project/Breast/InBreast/INBreast/AllTXTall/' # txtPath = 'F:/project/Breast/InBreast/INBreast/ALLtest/' outpathtxt = 'F:/project/Breast/InBreast/INBreast/outtxtpatch/' txtType = 'txt' txtLists = os.listdir(txtPath) #列出文件夹下所有的目录与文件 # Read all points into a list(before 21:30,715) for filename in txtLists: pp = 0 ppp = 0 with open(txtPath + filename, 'r') as file: print(filename) lines = file.readlines() dataset = [[] for i in range(len(lines))] for i in range(len(dataset)): dataset[i][:] = (item for item in lines[i].strip().split(',')) # 逐行读取数据 #print("dateset:", dataset) # print(type(dataset[0][0])) # print(dataset.__sizeof__()) u = np.array(dataset) list = np.zeros((u.shape[0], 2)) for i in range(u.shape[0]): # print(u[i,0][0]) findNumber = u[i,0].find(" ") # print(findNumber) list[i,0] = round(float(u[i, 0][0:findNumber])) list[i,1] = round(float(u[i, 0][findNumber+1:])) # Read the same name image(before 21:30,715) img = img_in_path + filename[:-4] + '.png' # out_path img = cv2.imread(img) black = np.zeros((img.shape[0], img.shape[1])) for i in range(img.shape[0]): for j in range(img.shape[1]): if img[i,j][0] == 0 and img[i,j][1] == 0 and img[i,j][2] == 0: black[i][j] = 1 print(img.shape[0],img.shape[1]) # Cutting distance has been determined, Handling cutting details(before 22.30, 715) row = img.shape[0]/56; col = img.shape[1]/56; print(row,col) # 待考虑问题:黑色像素 for i in range(int(row-1)): # print(i) for j in range(int(col-1)): tt = 0 pb = np.sum(black[i * 56:i * 56 + 112,j * 56:j * 56 + 112]) / 12544 # print("top_left:", i*56, j*56) # print("bottom_right:",i*56+112,j*56+112) for k in range(u.shape[0]): if j*56<list[k,0]<j*56+112 and i*56<list[k,1]<i*56+112: tt = tt + 1 patch_label = np.zeros((tt, 4)) tt1 = 0 for k in range(u.shape[0]): if j*56<list[k,0]<j*56+112 and i*56<list[k,1]<i*56+112: patch_label[tt1,0] = list[k,0] - j*56 patch_label[tt1,1] = list[k,1] - i*56 patch_label[tt1,2] = list[k,0] patch_label[tt1,3] = list[k,1] tt1 = tt1 + 1 if tt>0 and pb == 0: patch = img[i * 56:i * 56 + 112,j * 56:j * 56 + 112] pp = pp + 1 # print(patch.shape) outputimg = outpathimgyes + '/' + filename[:-4] + '/' + filename[:-4] + '_'+ str(pp) + '.png' cv2.imwrite(outputimg, patch) outputtxt1 = outpathtxt + '/' + filename[:-4] + '/' + filename[:-4] + '_' + str(pp) + '.txt' np.savetxt(outputtxt1, patch_label, fmt='%f') if tt==0 and pb == 0: patch = img[i * 56:i * 56 + 112,j * 56:j * 56 + 112] ppp = ppp + 1 # print(patch.shape) outputimg1 = outpathimgno + '/' + filename[:-4] + '/' + filename[:-4] + '_' + str(ppp) + '.png' cv2.imwrite(outputimg1, patch) print(img.shape[0], img.shape[1]) print(pp,ppp) for i in range(int(row-1)): tt = 0 pb = np.sum(black[i * 56:i * 56 + 112, img.shape[1] - 113:img.shape[1]-1]) / 12544 # print("top_right:",i*56,img.shape[1]-1) # print("bottom_left:", i * 56+112, img.shape[1] - 113) for k in range(u.shape[0]): if img.shape[1] - 113 < list[k, 0] < img.shape[1]-1 and i * 56 < list[k, 1] < i * 56 + 112: tt = tt + 1 patch_label = np.zeros((tt, 4)) tt1 = 0 # print(tt) for k in range(u.shape[0]): if img.shape[1] - 113 < list[k, 0] < img.shape[1]-1 and i * 56 < list[k, 1] < i * 56 + 112: patch_label[tt1, 0] = list[k, 0] - j * 56 patch_label[tt1, 1] = list[k, 1] - i * 56 patch_label[tt1, 2] = list[k, 0] patch_label[tt1, 3] = list[k, 1] tt1 = tt1 + 1 if tt > 0 and pb == 0.1: patch = img[i * 56:i * 56 + 112, img.shape[1] - 113:img.shape[1]-1] pp = pp + 1 # print(patch.shape) outputimg = outpathimgyes + '/' + filename[:-4] + '/' + filename[:-4] + '_' + str(pp) + '.png' cv2.imwrite(outputimg, patch) outputtxt1 = outpathtxt + '/' + filename[:-4] + '/' + filename[:-4] + '_' + str(pp) + '.txt' np.savetxt(outputtxt1, patch_label, fmt='%f') if tt == 0 and pb == 0: patch = img[i * 56:i * 56 + 112, img.shape[1] - 113:img.shape[1]-1] ppp = ppp + 1 # print(patch.shape) outputimg1 = outpathimgno + '/' + filename[:-4] + '/' + filename[:-4] + '_' + str(ppp) + '.png' cv2.imwrite(outputimg1, patch) print(img.shape[0], img.shape[1]) # # # for j in range(int(col-1)): # print("bottom_left:", img.shape[0]-1, j*56) # print("top_right:", img.shape[0] - 113, j * 56+112, ) tt = 0 pb = np.sum(black[img.shape[0]-113:img.shape[0] - 1, j * 56:j * 56+112]) / 12544 # print("top_right:",i*56,img.shape[1]-1) # print("bottom_left:", i * 56+112, img.shape[1] - 113) for k in range(u.shape[0]): if j * 56 < list[k, 0] < j * 56+112 and img.shape[0]-113 < list[k, 1] < img.shape[0]-1: tt = tt + 1 patch_label = np.zeros((tt, 4)) tt1 = 0 # print(tt) for k in range(u.shape[0]): if j * 56 < list[k, 0] < j * 56+112 and img.shape[0]-113 < list[k, 1] < img.shape[0]-1: patch_label[tt1, 0] = list[k, 0] - j * 56 patch_label[tt1, 1] = list[k, 1] - i * 56 patch_label[tt1, 2] = list[k, 0] patch_label[tt1, 3] = list[k, 1] tt1 = tt1 + 1 if tt > 0 and pb == 0: patch = img[img.shape[0]-113:img.shape[0] - 1, j * 56:j * 56+112] pp = pp + 1 # print(patch.shape) outputimg = outpathimgyes + '/' + filename[:-4] + '/' + filename[:-4] + '_' + str(pp) + '.png' cv2.imwrite(outputimg, patch) outputtxt1 = outpathtxt + '/' + filename[:-4] + '/' + filename[:-4] + '_' + str(pp) + '.txt' np.savetxt(outputtxt1, patch_label, fmt='%f') if tt == 0 and pb == 0: patch = img[img.shape[0]-113:img.shape[0] - 1, j * 56:j * 56+112] ppp = ppp + 1 # print(patch.shape) outputimg1 = outpathimgno + '/' + filename[:-4] + '/' + filename[:-4] + '_' + str(ppp) + '.png' cv2.imwrite(outputimg1, patch) # print(img.shape[0], img.shape[1]) # A = np.array([[1, 2], [3, 4], [5, 6]]) # print(np.sum(A)) print(pp)