Schleife durch die Knoten und Attribute extrahieren in Networkx

Ich in python definiert einige Formen mit den entsprechenden Eckpunkte, wie diese:

square = [[251, 184],
          [22, 192],
          [41, 350],
          [244, 346]]

triangle = [[250, 181],
            [133, 43],
            [21, 188]]

pentagon = [[131,  37],
            [11, 192],
            [37, 354],
            [247, 350],
            [256, 182]]

Dann nutze ich NetworkX-Paket um ein Diagramm zu erstellen:

G = nx.DiGraph()

Dann erstelle ich einen Knoten im Diagramm für jede Form:

G.add_node('square', points = square, center = (139, 265))
G.add_node('triangle', points = triangle, center = (139, 135))
G.add_node('pentagon', points = pentagon, center = (138, 223))

Nun ist das problem, ich haben Sie einige Kanten verbinden zwei Knoten, wenn eine Bedingung erfüllt ist. Die Bedingung zu erfüllen, ist wenn das Zentrum einer Form ist, die innerhalb oder außerhalb einer anderen Form, dann erstellen Sie einen Rand wie diese:

G.add_edge('triangle', 'pentagon', relation = 'inside')
G.add_edge('triangle', 'square', relation = 'outside')

Dazu habe ich eine Schleife über das Knoten, extrahieren Sie die center einer Form, extrahieren Sie die points des andere Formen (NICHT selbst, es ist nutzlos) und machen die pointPolygonTest.

Ich versuche schon Recht viel, aber nicht kam mit einer Lösung. Die nächsten (nicht wirklich effektiv) Lösung, die ich habe ist dieses:

nodes_p=dict([((u),d['points']) for u,d in G.nodes(data=True)])
nodes_c=dict([((u),d['center']) for u,d in G.nodes(data=True)])
for z,c in nodes_c.items():
    print z + ' with center', c
    for z,p in nodes_p.items():
        p_array = np.asarray(p)
        if cv2.pointPolygonTest(p_array,c,False)>=0:
            print 'inside ' + z
            #create edge
        else:
            print 'outside ' + z
            #create edge

Dieser gibt mir folgende Ausgabe, das ist nicht optimal, weil es eine Beziehung, die hätte vermieden werden können (wie triangle inside triangle) oder falsche Beziehungen (wie pentagon inside square)

triangle with center (139, 135)
inside triangle
outside square
inside pentagon
square with center (139, 265)
outside triangle
inside square
inside pentagon
pentagon with center (138, 223)
outside triangle
inside square
inside pentagon

Wie kann ich dieses problem lösen? Jede Anregung ist apreciated. Zur Erinnerung: das Hauptproblem ist, wie eine Schleife durch die Knoten, und extrahieren Sie die info. Die Pakete, die ich importieren, die für das gesamte Skript sind:

import numpy as np
import networkx as nx
import cv2

Sie sollten wahrscheinlich hinzufügen Ihre import-Anweisungen (numpy und cv2) zu helfen, die künftigen Leser

InformationsquelleAutor Francesco Sgaramella | 2014-02-07

6

Hier ist ein Bild von Ihrem Polygone

Ersten, es gibt keine Notwendigkeit zu werfen, die die Knoten wie Wörterbücher, können wir die Iteration auf Sie direkt. Dieser code ist aus der Basis von dieses Beispiel
```
for u,outer_d in G.nodes(data=True):
   center = outer_d['center']
   print u, "with center", center
   for v, inner_d in G.nodes(data=True):
        #Don't compare self to self
        if u != v:
            # Create a source image
            src = np.zeros((400,400),np.uint8)          
            # draw an polygon on image src
            points = np.array(inner_d['points'],np.int0)
            cv2.polylines(src,[points],True,255,3)
            contours,_ = cv2.findContours(src,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
            if cv2.pointPolygonTest(contours[0],center,True) <= 0:
                print 'outside',v
            else:
                print 'inside',v
```
Die Ausgabe ist
```
pentagon with center (138, 223)
inside square
outside triangle
square with center (139, 265)
inside pentagon
outside triangle
triangle with center (139, 135)
inside pentagon
outside square
```
Da das Ziel ist zu bestimmen, ob ein polygon ist vollständig in die anderen, sollten wir prüfen alle vertices eines Polygons in einer anderen. Hier ist eine vorläufige (leider nicht getestet) - Lösung.
```
def checkPoint(point, poly,r=400):
    ''' determine if point is on the interior of poly'''
    # Create a source image
    src = np.zeros((r,r),np.uint8)       
    # draw an polygon on image src
    verts = np.array(poly,np.int0)
    cv2.polylines(src,[verts],True,255,3)
    contours,_ = cv2.findContours(src,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
    return cv2.pointPolygonTest(contours[0],tuple(point),True) > 0:


for u,outer_d in G.nodes(data=True):
    points = outer_d['points']
    center = outer_d['center']
    print u, "with center", center
    for v, inner_d in G.nodes(data=True):
        poly = inner_d['points']
        if u != v:
            if all([checkPoint(point,poly) for point in points]):
                print 'inside',v
            else:
                print 'outside',v
```
Die Ausgabe für dieses Beispiel wie folgt und sollten nun korrekt sein.
```
pentagon with center (138, 223)
outside square
outside triangle
square with center (139, 265)
inside pentagon
outside triangle
triangle with center (139, 135)
inside pentagon
outside square
```
Beachten Sie, dass ich gemacht habe, die Annahme, dass die Polygone konvex sein. Wenn dies nicht der Fall, dann könnten Sie prüfen alle Punkte auf der Kontur, anstatt nur in der Ecke Punkten. Sie können auch bauen in einer Konvexität überprüfen mit cv2 finden Sie dieser blog für details.
- Sorry, verwechselt, die die Ungleichheit. Ist es jetzt korrekt?
- Die Ausgabe ist falsch. Es sollte sein: pentagon außerhalb Dreieck, außen Quadrat, Quadrat im inneren pentagon, außerhalb Dreieck, Dreieck innerhalb pentagon, außerhalb Quadrat
- "innen" bedeutet die Mitte drin ist, richtig? Wenn dem so ist, sollte pentagon außerhalb Dreieck, innerhalb Platz, Platz im inneren pentagon, außerhalb Dreieck, Dreieck innerhalb pentagon, außerhalb Quadrat
- Was du sagst, ist tatsächlich wahr. Jedoch ist nicht konsistent mit dem Bild bin ich mit dem Experimentieren. In der Bild gibt es ein pentagon mit einem Quadrat und einem Dreieck. Der Platz ist unter dem Dreieck. So, dass ist der Grund, warum die Ausgabe ist nicht 100% perfekt. Ich weiß nicht, wie es zu lösen. Da natürlich auch die pointPolygonTest für das pentagon gegen Platz zurück hinein, doch sollte tatsächlich draußen.
- Das image, das Sie vollkommen wahr ist. So können Sie sehen, dass der Schwerpunkt des pentagon befindet sich innerhalb des Platzes, aber tatsächlich ist das pentagon enthält das Quadrat, was bedeutet, dass das pentagon, außerhalb des Quadrats. Wie dies zu überwinden?
- Ok, eine Lösung wäre zu überprüfen, um zu sehen, ob jedes point ein polygon innerhalb der anderen. Ich will hinzufügen, eine vorläufige Lösung, aber ich kann es nicht testen, da ich jetzt nicht cv2 auf meinem computer in der Arbeit.
- Es ist nicht Ihre Lösung. Aber ich Verstand, was Sie versuchen, und ich werde versuchen mich. Ich lasse Sie wissen, wie es geht. Jedenfalls wirklich vielen Dank für Eure Hilfe!
- Ich habe erkannt und behoben, die Probleme mit der zweiten Lösung. Hinweis: die tuple in der letzten Zeile der Funktion und der [ und ] im Aufruf all. Ich hoffe, das hilft.
InformationsquelleAutor yardsale8

Schreibe einen Kommentar

Du musst angemeldet sein, um einen Kommentar abzugeben.