def __createMap(self, xmlfile): """根据配置文件中map映射关系,建立各个资源的连接情况""" def nameScan(rs, name): for r in rs: if r.name == name: return r return None def fullNameScan(rs, fullname): for name in fullname.split(":"): ret = nameScan(rs, name) if ret == None: break else: rs = ret.children return ret node = Parse().getElementTreeRoot(xmlfile).find("resources_map") for con in node.findall("con"): r1,r2 = fullNameScan(self.children, con.get("con1")),fullNameScan(self.children, con.get("con2")) r1.connect.append(r2) r2.connect.append(r1)
old_center_x, old_center_y = px + pw / 2.0, py + ph / 2.0 x_ping, y_ping, w_suo, h_suo = R_result[0][1], \ R_result[0][2], \ R_result[0][3], \ R_result[0][4] new__center_x = x_ping * pw + old_center_x new__center_y = y_ping * ph + old_center_y new_w = pw * np.exp(w_suo.data.cpu().numpy()) new_h = ph * np.exp(h_suo.data.cpu().numpy()) a = new__center_x.data.cpu().numpy() b = new__center_y.data.cpu().numpy() a = float(a) b = float(b) new_verts = [a, b, new_w, new_h] results.append(new_verts) results_label.append(pred[0]) print('find proposal, label:', pred[0], "rect:", new_verts) count += 1 def NMS(self): pass if __name__ == '__main__': parse = Parse() config = parse.config image = cv2.imread(config.test_example) rcnn = RCNN(config=config) rcnn.test(image)
def __registChildren(self, xmlfile): node = Parse().getElementTreeRoot(xmlfile).find("resources") self.addChild(map(self.libSearch, node.getchildren()))