def getData(): index=1 '''报错数据集有(空心):han,jakobs1,jakobs2 ''' '''形状过多暂时未处理:shapes、shirt、swim、trousers''' name=["ga","albano","blaz1","blaz2","dighe1","dighe2","fu","han","jakobs1","jakobs2","mao","marques","shapes","shirts","swim","trousers","convex"] print("开始处理",name[index],"数据集") # ga为测试数据集 '''暂时没有考虑宽度,全部缩放来表示''' scale=[100,0.25,100,100,20,20,20,10,20,20,0.5,20,50,1,1,1,1,1,1,1] print("缩放",scale[index],"倍") df = pd.read_csv("/Users/sean/Documents/Projects/Packing-Algorithm/euro_data/"+name[index]+".csv") polygons=[] for i in range(0,df.shape[0]): for j in range(0,df['num'][i]): poly=json.loads(df['polygon'][i]) GeoFunc.normData(poly,scale[index]) polygons.append(poly) return getDecreasing(polygons)
def combinateTestData(self): shapes = pd.read_csv("/Users/sean/Documents/Projects/Packing-Algorithm/euro_data/train_test/blaz_vec.csv") for i in range(0,60): index=[random.randint(0,6),random.randint(0,6),random.randint(0,6),random.randint(0,6)] shape1=json.loads(shapes["poly"][index[0]]) shape2=json.loads(shapes["poly"][index[1]]) shape3=json.loads(shapes["poly"][index[2]]) shape4=json.loads(shapes["poly"][index[3]]) GeoFunc.normData(shape1,150) GeoFunc.normData(shape2,150) GeoFunc.normData(shape3,150) GeoFunc.normData(shape4,150) with open("/Users/sean/Documents/Projects/Packing-Algorithm/euro_data/train_test/blaz_trian_test.csv","a+") as csvfile: writer = csv.writer(csvfile) x_8=self.normVec(json.loads(shapes["vec_8"][index[0]])+json.loads(shapes["vec_8"][index[1]])+json.loads(shapes["vec_8"][index[2]])+json.loads(shapes["vec_8"][index[3]])) x_16=self.normVec(json.loads(shapes["vec_16"][index[0]])+json.loads(shapes["vec_16"][index[1]])+json.loads(shapes["vec_16"][index[2]])+json.loads(shapes["vec_16"][index[3]])) x_32=self.normVec(json.loads(shapes["vec_32"][index[0]])+json.loads(shapes["vec_32"][index[1]])+json.loads(shapes["vec_32"][index[2]])+json.loads(shapes["vec_32"][index[3]])) x_64=self.normVec(json.loads(shapes["vec_64"][index[0]])+json.loads(shapes["vec_64"][index[1]])+json.loads(shapes["vec_64"][index[2]])+json.loads(shapes["vec_64"][index[3]])) x_128=self.normVec(json.loads(shapes["vec_128"][index[0]])+json.loads(shapes["vec_128"][index[1]])+json.loads(shapes["vec_128"][index[2]])+json.loads(shapes["vec_128"][index[3]])) x_256=self.normVec(json.loads(shapes["vec_256"][index[0]])+json.loads(shapes["vec_256"][index[1]])+json.loads(shapes["vec_256"][index[2]])+json.loads(shapes["vec_256"][index[3]])) writer.writerows([[i,time.asctime(time.localtime(time.time())),index[0],index[1],index[2],index[3],shape1,shape2,shape3,shape4,x_8,x_16,x_32,x_64,x_128,x_256]])