class SNDG_APP(QtWidgets.QMainWindow):
def __init__(self, args, parent=None):
QtWidgets.QMainWindow.__init__(self, parent)
self.world = None
self.model = None
self.savePix = False
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.ui.graphicsView.show()
self.ui.graphicsView.setRenderHint(QtGui.QPainter.Antialiasing)
self.ui.graphicsView.setViewportUpdateMode(QtWidgets.QGraphicsView.BoundingRectViewportUpdate)
self.labels = [ self.ui.label0, self.ui.label1, self.ui.label2]
self.dataset = None
if len(args) == 1:
if args[0].endswith('.json'):
self.dataset = open(args[0], 'r').readlines() # Get all the dataset including duplicates
random.shuffle(self.dataset)
else:
print('The dataset file name should be a json file (check extension).')
sys.exit(-1)
elif len(args)>1:
print('The **optional** dataset file name "dataset.json" should be the only parameter.')
sys.exit(-1)
@QtCore.Slot(int)
def on_slider_valueChanged(self, value):
value = float(len(self.labels)-1) * float(value)/float(self.ui.slider.maximum()-self.ui.slider.minimum())
for i, label in zip(range(len(self.labels)), self.labels):
v = 1.-math.fabs(i-value)
if v < 0: v = 0
label.setStyleSheet('color: rgba(0, 0, 0, {});'.format(v))
self.ui.sendButton.setEnabled(True)
@QtCore.Slot()
def on_sendButton_clicked(self):
textVal = str(self.ui.slider.value()).zfill(3)
self.ui.statusbar.showMessage("send" + textVal)
self.ui.sendButton.setEnabled(False)
s= self.world.serialize(self.ui.slider.value())
if self.savePix:
myPixmap = QtWidgets.QWidget.grab(self.ui.graphicsView)
myPixmap.save(s['identifier']+".png")
self.on_getButton_clicked()
@QtCore.Slot()
def on_getButton_clicked(self):
self.populateWorld()
if self.ui.estimateBox.isChecked():
self.on_estimateButton_clicked()
def generateDataset(self, number, savePix=False):
self.savePix = savePix
for i in range(number):
self.on_sendButton_clicked()
@QtCore.Slot()
def on_estimateButton_clicked(self):
import torch
import pickle
import gat
# import gcn
self.ui.statusbar.showMessage("estimate")
if self.model is None:
params = pickle.load(open('model.prms', 'rb'))
# self.model = gcn.GCN(*params)
self.model = gat.GAT(*params)
self.model.load_state_dict(torch.load('model.tch'))
self.model.eval()
device = torch.device("cpu")
structure = self.world.serialize()
train_dataset = socnav.SocNavDataset(structure, mode='train')
train_dataloader = DataLoader(train_dataset, batch_size=1, collate_fn=collate)
for batch, data in enumerate(train_dataloader):
subgraph, feats, labels = data
feats = feats.to(device)
self.model.g = subgraph
# for layer in self.model.layers:
for layer in self.model.gat_layers:
layer.g = subgraph
logits = self.model(feats.float())[0].detach().numpy()[0]
translate = 1. / (1. + math.exp(-logits))*100
if translate < 0: translate = 0
if translate > 100: translate = 100
self.ui.slider.setValue(int(translate))
def closeEvent(self, event):
if self.dataset:
fd = open('saved.json', 'w')
fd.write(self.current_line)
for x in self.dataset:
fd.write(x)
fd.close()
event.accept()
def eventFilter(self, receiver, event):
if event.type() is not QtCore.QEvent.Type.KeyRelease:
return super(SNDG_APP, self).eventFilter(receiver, event)
elif event.key() == 16777235:
self.ui.slider.setValue(self.ui.slider.value()+1)
elif event.key() == 16777238:
self.ui.slider.setValue(self.ui.slider.value()+7)
elif event.key() == 16777239:
self.ui.slider.setValue(self.ui.slider.value()-7)
elif event.key() == 16777237:
self.ui.slider.setValue(self.ui.slider.value()-1)
elif event.key() == 16777220:
if self.ui.sendButton.isEnabled():
self.on_sendButton_clicked()
elif event.key() == QtCore.Qt.Key_Home or event.key() == QtCore.Qt.Key_5:
self.ui.slider.setValue(self.ui.slider.maximum())
elif event.key() == QtCore.Qt.Key_End or event.key() == QtCore.Qt.Key_1:
self.ui.slider.setValue(self.ui.slider.minimum())
elif event.key() == QtCore.Qt.Key_2:
self.ui.slider.setValue(self.ui.slider.maximum()/4*1)
elif event.key() == QtCore.Qt.Key_3:
self.ui.slider.setValue(self.ui.slider.maximum() / 4 * 2)
elif event.key() == QtCore.Qt.Key_4:
self.ui.slider.setValue(self.ui.slider.maximum() / 4 * 3)
event.accept()
return True
def populateWorld(self):
if self.dataset is None:
self.world = WorldGenerator()
else:
try:
self.current_line = self.dataset.pop(0)
except IndexError:
sys.exit(0)
self.world = WorldGenerator(self.current_line)
self.ui.graphicsView.setScene(self.world)
self.ui.sendButton.setEnabled(False)
class SNDG_APP(QtWidgets.QMainWindow):
def __init__(self, args, parent=None):
QtWidgets.QMainWindow.__init__(self, parent)
self.world = None
self.model = None
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.ui.graphicsView.show()
self.ui.graphicsView.setRenderHint(QtGui.QPainter.Antialiasing)
self.ui.graphicsView.setViewportUpdateMode(
QtWidgets.QGraphicsView.BoundingRectViewportUpdate)
self.labels = [
self.ui.label0, self.ui.label1, self.ui.label2, self.ui.label3,
self.ui.label4, self.ui.label5
]
self.dataset = None
if len(args) == 1:
if args[0].endswith('.json'):
self.dataset = open(args[0], 'r').readlines(
) # Get all the dataset including duplicates
random.shuffle(self.dataset)
else:
print(
'The dataset file name should be a json file (check extension).'
)
sys.exit(-1)
elif len(args) > 1:
print(
'The **optional** dataset file name "dataset.json" should be the only parameter.'
)
sys.exit(-1)
@QtCore.Slot(int)
def on_slider_valueChanged(self, value):
value = float(len(self.labels) -
1) * float(value) / float(self.ui.slider.maximum() -
self.ui.slider.minimum())
for i, label in zip(range(len(self.labels)), self.labels):
v = 1. - math.fabs(i - value)
if v < 0: v = 0
label.setStyleSheet('color: rgba(0, 0, 0, {});'.format(v))
self.ui.sendButton.setEnabled(True)
@QtCore.Slot()
def on_sendButton_clicked(self):
textVal = str(self.ui.slider.value()).zfill(3)
self.ui.statusbar.showMessage("send" + textVal)
self.ui.sendButton.setEnabled(False)
self.world.serialize(self.ui.slider.value())
self.on_getButton_clicked()
@QtCore.Slot()
def on_getButton_clicked(self):
self.populateWorld()
if self.ui.estimateBox.isChecked():
self.on_estimateButton_clicked()
def generateDataset(self, number):
for i in range(number):
self.on_sendButton_clicked()
@QtCore.Slot()
def on_estimateButton_clicked(self):
self.ui.statusbar.showMessage("estimate")
if self.model is None:
self.device = torch.device('cpu')
self.params = pickle.load(open('../SNGNN_PARAMETERS.prms', 'rb'),
fix_imports=True)
self.GNNmodel = pg_rgcn_gat.PRGAT(self.params[5],
self.params[7],
self.params[8][0],
self.params[14],
self.params[14],
self.params[6],
int(self.params[4] / 2),
int(self.params[4] / 2),
self.params[10],
self.params[9],
self.params[12],
bias=True)
self.GNNmodel.load_state_dict(
torch.load('../SNGNN_MODEL.tch', map_location='cpu'))
self.GNNmodel.eval()
print(json.loads(self.current_line))
graph_type = 'relational'
train_dataset = socnav.SocNavDataset(json.loads(self.current_line),
mode='train',
alt=graph_type,
verbose=False)
train_dataloader = DataLoader(train_dataset,
batch_size=1,
collate_fn=collate)
for batch, data in enumerate(train_dataloader):
subgraph, feats, labels = data
feats = feats.to(self.device)
data = Data(
x=feats.float(),
edge_index=torch.stack(subgraph.edges()).to(self.device),
edge_type=subgraph.edata['rel_type'].squeeze().to(self.device))
logits = self.GNNmodel(data)[0].detach().numpy()[0]
score = logits * 100
if score > 100:
score = 100
elif score < 0:
score = 0
self.ui.slider.setValue(int(score))
def closeEvent(self, event):
if self.dataset:
fd = open('saved.json', 'w')
fd.write(self.current_line)
for x in self.dataset:
fd.write(x)
fd.close()
event.accept()
def eventFilter(self, receiver, event):
if event.type() is not QtCore.QEvent.Type.KeyRelease:
return super(SNDG_APP, self).eventFilter(receiver, event)
elif event.key() == 16777235:
self.ui.slider.setValue(self.ui.slider.value() + 1)
elif event.key() == 16777238:
self.ui.slider.setValue(self.ui.slider.value() + 7)
elif event.key() == 16777239:
self.ui.slider.setValue(self.ui.slider.value() - 7)
elif event.key() == 16777237:
self.ui.slider.setValue(self.ui.slider.value() - 1)
elif event.key() == 16777220:
if self.ui.sendButton.isEnabled():
self.on_sendButton_clicked()
elif event.key() == QtCore.Qt.Key_Home or event.key(
) == QtCore.Qt.Key_5:
self.ui.slider.setValue(self.ui.slider.maximum())
elif event.key() == QtCore.Qt.Key_End or event.key(
) == QtCore.Qt.Key_1:
self.ui.slider.setValue(self.ui.slider.minimum())
elif event.key() == QtCore.Qt.Key_2:
self.ui.slider.setValue(self.ui.slider.maximum() / 4 * 1)
elif event.key() == QtCore.Qt.Key_3:
self.ui.slider.setValue(self.ui.slider.maximum() / 4 * 2)
elif event.key() == QtCore.Qt.Key_4:
self.ui.slider.setValue(self.ui.slider.maximum() / 4 * 3)
event.accept()
return True
def populateWorld(self):
if self.dataset is None:
self.world = WorldGenerator()
self.current_line = json.dumps(self.world.serialize())
else:
try:
self.current_line = self.dataset.pop(0)
except IndexError:
sys.exit(0)
self.world = WorldGenerator(self.current_line)
self.ui.graphicsView.setScene(self.world)
self.ui.sendButton.setEnabled(False)