Merge pull request '222' (#14) from mch_branch into master
commit
941aa7730b
|
After Width: | Height: | Size: 32 KiB |
|
After Width: | Height: | Size: 23 KiB |
|
After Width: | Height: | Size: 23 KiB |
|
After Width: | Height: | Size: 36 KiB |
|
After Width: | Height: | Size: 29 KiB |
|
After Width: | Height: | Size: 9.0 KiB |
|
After Width: | Height: | Size: 14 KiB |
@ -0,0 +1,2 @@
|
||||
# work1
|
||||
|
||||
@ -0,0 +1,99 @@
|
||||
import os
|
||||
import tensorflow as tf
|
||||
train_dataset_url = "https://storage.googleapis.com/download.tensorflow.org/data/iris_training.csv"
|
||||
file_url = "file:///home/smj/Work/mhand/test.csv"
|
||||
train_dataset_fp = tf.keras.utils.get_file(fname=os.path.basename(train_dataset_url),
|
||||
origin=train_dataset_url)
|
||||
|
||||
print("Local copy of the dataset file: {}".format(train_dataset_fp))
|
||||
# column order in CSV file
|
||||
|
||||
class_names = ['Iris setosa', 'Iris versicolor', 'Iris virginica']
|
||||
|
||||
batch_size = 32
|
||||
|
||||
train_dataset = tf.data.experimental.make_csv_dataset(
|
||||
train_dataset_fp,
|
||||
batch_size,
|
||||
label_name="virginica",
|
||||
num_epochs=1)
|
||||
features, labels = next(iter(train_dataset))
|
||||
|
||||
|
||||
def pack_features_vector(features, labels):
|
||||
"""Pack the features into a single array."""
|
||||
features = tf.stack(list(features.values()), axis=1)
|
||||
return features, labels
|
||||
|
||||
train_dataset = train_dataset.map(pack_features_vector)
|
||||
features, labels = next(iter(train_dataset))
|
||||
print(features)
|
||||
|
||||
model = tf.keras.Sequential([
|
||||
tf.keras.layers.Dense(10, activation=tf.nn.relu, input_shape=(4,)), # input shape required
|
||||
tf.keras.layers.Dense(10, activation=tf.nn.relu),
|
||||
tf.keras.layers.Dense(3)
|
||||
])
|
||||
# features[0].shape=(4,0)
|
||||
predictions = model(features)
|
||||
tf.nn.softmax(predictions[:5])
|
||||
loss_object = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
|
||||
|
||||
def loss(model, x, y, training):
|
||||
# training=training is needed only if there are layers with different
|
||||
# behavior during training versus inference (e.g. Dropout).
|
||||
y_ = model(x, training=training)
|
||||
|
||||
return loss_object(y_true=y, y_pred=y_)
|
||||
|
||||
|
||||
l = loss(model, features, labels, training=False)
|
||||
def grad(model, inputs, targets):
|
||||
with tf.GradientTape() as tape:
|
||||
loss_value = loss(model, inputs, targets, training=True)
|
||||
return loss_value, tape.gradient(loss_value, model.trainable_variables)
|
||||
|
||||
optimizer = tf.keras.optimizers.SGD(learning_rate=0.01)
|
||||
loss_value, grads = grad(model, features, labels)
|
||||
|
||||
print("Step: {}, Initial Loss: {}".format(optimizer.iterations.numpy(),
|
||||
loss_value.numpy()))
|
||||
|
||||
optimizer.apply_gradients(zip(grads, model.trainable_variables))
|
||||
|
||||
print("Step: {}, Loss: {}".format(optimizer.iterations.numpy(),
|
||||
loss(model, features, labels, training=True).numpy()))
|
||||
|
||||
## Note: Rerunning this cell uses the same model variables
|
||||
|
||||
# Keep results for plotting
|
||||
train_loss_results = []
|
||||
train_accuracy_results = []
|
||||
|
||||
num_epochs = 201
|
||||
|
||||
for epoch in range(num_epochs):
|
||||
epoch_loss_avg = tf.keras.metrics.Mean()
|
||||
epoch_accuracy = tf.keras.metrics.SparseCategoricalAccuracy()
|
||||
|
||||
# Training loop - using batches of 32
|
||||
for x, y in train_dataset:
|
||||
# Optimize the model
|
||||
loss_value, grads = grad(model, x, y)
|
||||
optimizer.apply_gradients(zip(grads, model.trainable_variables))
|
||||
|
||||
# Track progress
|
||||
epoch_loss_avg.update_state(loss_value) # Add current batch loss
|
||||
# Compare predicted label to actual label
|
||||
# training=True is needed only if there are layers with different
|
||||
# behavior during training versus inference (e.g. Dropout).
|
||||
epoch_accuracy.update_state(y, model(x, training=True))
|
||||
|
||||
# End epoch
|
||||
train_loss_results.append(epoch_loss_avg.result())
|
||||
train_accuracy_results.append(epoch_accuracy.result())
|
||||
|
||||
if epoch % 50 == 0:
|
||||
print("Epoch {:03d}: Loss: {:.3f}, Accuracy: {:.3%}".format(epoch,
|
||||
epoch_loss_avg.result(),
|
||||
epoch_accuracy.result()))
|
||||
@ -0,0 +1,153 @@
|
||||
<!DOCTYPE html>
|
||||
<html>
|
||||
<head>
|
||||
<meta charset="utf-8">
|
||||
<script src="/static/hand/camera_utils.js" crossorigin="anonymous"></script>
|
||||
<script src="/static/hand/control_utils.js" crossorigin="anonymous"></script>
|
||||
<script src="/static/hand/drawing_utils.js" crossorigin="anonymous"></script>
|
||||
<script src="/static/hand/hands.js" crossorigin="anonymous"></script>
|
||||
</head>
|
||||
|
||||
<body>
|
||||
<div class="container">
|
||||
<canvas class="output_canvas" width="1280px" height="720px"></canvas>
|
||||
</div>
|
||||
<div id="txtHint"></div>
|
||||
<script type="module">
|
||||
function ajax(str) {
|
||||
var xmlhttp;
|
||||
if (str.length==0)
|
||||
{
|
||||
document.getElementById("txtHint").innerHTML="";
|
||||
return;
|
||||
}
|
||||
if (window.XMLHttpRequest) xmlhttp=new XMLHttpRequest();
|
||||
else xmlhttp=new ActiveXObject("Microsoft.XMLHTTP");
|
||||
xmlhttp.onreadystatechange=function()
|
||||
{
|
||||
if (xmlhttp.readyState==4 && xmlhttp.status==200)
|
||||
{
|
||||
document.getElementById("txtHint").innerHTML+=xmlhttp.responseText;
|
||||
num=0
|
||||
}
|
||||
}
|
||||
xmlhttp.open("GET","/trans?="+str,true)
|
||||
xmlhttp.send();
|
||||
}
|
||||
|
||||
const videoElement = document.createElement("video");
|
||||
const canvasElement = document.getElementsByClassName('output_canvas')[0];
|
||||
const canvasCtx = canvasElement.getContext('2d');
|
||||
var str =""
|
||||
var pre=str
|
||||
var num=2
|
||||
function onResults(results) {
|
||||
canvasCtx.save();
|
||||
canvasCtx.clearRect(0, 0, canvasElement.width, canvasElement.height);
|
||||
canvasCtx.drawImage(results.image, 0, 0, canvasElement.width, canvasElement.height);
|
||||
if (results.multiHandLandmarks) {
|
||||
var Left=[0,0,0,0,0,0,0,0,0]
|
||||
var Right=[0,0,0,0,0,0,0,0,0]
|
||||
for (let i = 0; i < results.multiHandLandmarks.length; i++) {
|
||||
var hand_lable=results.multiHandedness[i].label
|
||||
if (hand_lable=="Left"){
|
||||
Left[0]=1
|
||||
var wrist=results.multiHandLandmarks[i][0]
|
||||
var index_finger_mcp=results.multiHandLandmarks[i][5]
|
||||
var index_finger_tip=results.multiHandLandmarks[i][8]
|
||||
var middle_finger_mcp=results.multiHandLandmarks[i][9]
|
||||
var middle_finger_tip=results.multiHandLandmarks[i][12]
|
||||
var ring_finger_mcp=results.multiHandLandmarks[i][13]
|
||||
var ring_finger_tip=results.multiHandLandmarks[i][16]
|
||||
var pinky_mcp=results.multiHandLandmarks[i][17]
|
||||
var pinky_tip=results.multiHandLandmarks[i][20]
|
||||
|
||||
var one_x = index_finger_mcp.x-pinky_mcp.x
|
||||
var one_y = index_finger_mcp.y-pinky_mcp.y
|
||||
|
||||
if (Math.abs(one_x)>Math.abs(one_y)) {
|
||||
Left[3]=1;//平
|
||||
if (one_x>0) Left[1]=1;//正
|
||||
var tow = index_finger_mcp.y-wrist.y;
|
||||
if (tow<0) Left[2]=1;//上
|
||||
var length = Math.abs(middle_finger_mcp.y-wrist.y)/3;
|
||||
if (Math.abs(index_finger_tip.y-index_finger_mcp.y)>length) Left[5]=1;
|
||||
if (Math.abs(middle_finger_tip.y-middle_finger_mcp.y)>length) Left[6]=1;
|
||||
if (Math.abs(ring_finger_tip.y-ring_finger_mcp.y)>length) Left[7]=1;
|
||||
if (Math.abs(pinky_tip.y-pinky_mcp.y)>length) Left[8]=1;
|
||||
} else {
|
||||
var length = Math.abs(one_y)/3;
|
||||
if (Math.abs(index_finger_mcp.x-index_finger_tip.x)>length) Left[5]=1;
|
||||
if (Math.abs(middle_finger_mcp.x-middle_finger_tip.x)>length) Left[6]=1;
|
||||
if (Math.abs(ring_finger_mcp.x-ring_finger_tip.x)>length) Left[7]=1;
|
||||
if (Math.abs(pinky_mcp.x-pinky_tip.x)>length) Left[8]=1;
|
||||
}
|
||||
} else {
|
||||
Right[0]=1
|
||||
var wrist=results.multiHandLandmarks[i][0]
|
||||
var index_finger_mcp=results.multiHandLandmarks[i][5]
|
||||
var index_finger_tip=results.multiHandLandmarks[i][8]
|
||||
var middle_finger_mcp=results.multiHandLandmarks[i][9]
|
||||
var middle_finger_tip=results.multiHandLandmarks[i][12]
|
||||
var ring_finger_mcp=results.multiHandLandmarks[i][13]
|
||||
var ring_finger_tip=results.multiHandLandmarks[i][16]
|
||||
var pinky_mcp=results.multiHandLandmarks[i][17]
|
||||
var pinky_tip=results.multiHandLandmarks[i][20]
|
||||
|
||||
var one_x = index_finger_mcp.x-pinky_mcp.x
|
||||
var one_y = index_finger_mcp.y-pinky_mcp.y
|
||||
if (Math.abs(one_x)>Math.abs(one_y)) {
|
||||
Right[3]=1;//平
|
||||
if (one_x>0) Right[1]=1;//正
|
||||
var tow = index_finger_mcp.y-wrist.y;
|
||||
if (tow<0) Right[2]=1;//上
|
||||
var length = Math.abs(middle_finger_mcp.y-wrist.y)/3;
|
||||
if (Math.abs(index_finger_tip.y-index_finger_mcp.y)>length) Right[5]=1;
|
||||
if (Math.abs(middle_finger_tip.y-middle_finger_mcp.y)>length) Right[6]=1;
|
||||
if (Math.abs(ring_finger_tip.y-ring_finger_mcp.y)>length) Right[7]=1;
|
||||
if (Math.abs(pinky_tip.y-pinky_mcp.y)>length) Right[8]=1;
|
||||
} else {
|
||||
var length = Math.abs(one_y)/3;
|
||||
if (Math.abs(index_finger_mcp.x-index_finger_tip.x)>length) Right[5]=1;
|
||||
if (Math.abs(middle_finger_mcp.x-middle_finger_tip.x)>length) Right[6]=1;
|
||||
if (Math.abs(ring_finger_mcp.x-ring_finger_tip.x)>length) Right[7]=1;
|
||||
if (Math.abs(pinky_mcp.x-pinky_tip.x)>length) Right[8]=1;
|
||||
}
|
||||
}
|
||||
pre=str
|
||||
str=''
|
||||
for (let j = 0; j < Left.length; j++) str+=Left[j]
|
||||
str+='|'
|
||||
for (let j = 0; j < Right.length; j++) str+=Right[j]
|
||||
if (pre==str) num=num+1
|
||||
console.log(pre,str,num)
|
||||
if (num==20) ajax(str)
|
||||
drawConnectors(canvasCtx, results.multiHandLandmarks[i], HAND_CONNECTIONS,
|
||||
{color: '#00FF00', lineWidth: 5});
|
||||
drawLandmarks(canvasCtx, results.multiHandLandmarks[i], {color: '#FF0000', lineWidth: 2});
|
||||
}
|
||||
}
|
||||
canvasCtx.restore();
|
||||
}
|
||||
|
||||
const hands = new Hands({locateFile: (file) => {
|
||||
return `/static/hand/${file}`;
|
||||
}});
|
||||
hands.setOptions({
|
||||
maxNumHands: 2,
|
||||
modelComplexity: 1,
|
||||
minDetectionConfidence: 0.5,
|
||||
minTrackingConfidence: 0.5
|
||||
});
|
||||
hands.onResults(onResults);
|
||||
const camera = new Camera(videoElement, {
|
||||
onFrame: async () => {
|
||||
await hands.send({image: videoElement});
|
||||
},
|
||||
width: 1280,
|
||||
height: 720
|
||||
});
|
||||
camera.start();
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
Binary file not shown.
Binary file not shown.
@ -0,0 +1,139 @@
|
||||
import math
|
||||
import sqlite3
|
||||
|
||||
print ("数据库打开成功")
|
||||
strs = "INSERT INTO XYZ VALUES ("
|
||||
|
||||
|
||||
import cv2
|
||||
import mediapipe as mp
|
||||
|
||||
mp_drawing = mp.solutions.drawing_utils
|
||||
mp_drawing_styles = mp.solutions.drawing_styles
|
||||
mp_hands = mp.solutions.hands
|
||||
|
||||
num = 0
|
||||
# For webcam input:
|
||||
cap = cv2.VideoCapture(0)
|
||||
Left = [0, 0, 0, 0, 0, 0, 0, 0, 0]
|
||||
Right = [0, 0, 0, 0, 0, 0, 0, 0, 0]
|
||||
|
||||
with mp_hands.Hands(
|
||||
model_complexity=0,
|
||||
max_num_hands=2,
|
||||
min_detection_confidence=0.5,
|
||||
min_tracking_confidence=0.5) as hands:
|
||||
while cap.isOpened():
|
||||
success, image = cap.read()
|
||||
if not success:
|
||||
print("Ignoring empty camera frame.")
|
||||
# If loading a video, use 'break' instead of 'continue'.
|
||||
continue
|
||||
|
||||
# To improve performance, optionally mark the image as not writeable to
|
||||
# pass by reference.
|
||||
image.flags.writeable = False
|
||||
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
|
||||
results = hands.process(image)
|
||||
|
||||
# Draw the hand annotations on the image.
|
||||
image.flags.writeable = True
|
||||
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
|
||||
aa=0
|
||||
if results.multi_hand_landmarks:
|
||||
for i,hand_landmarks in enumerate(results.multi_hand_landmarks):
|
||||
Left = [0, 0, 0, 0, 0, 0, 0, 0, 0]
|
||||
Right = [0, 0, 0, 0, 0, 0, 0, 0, 0]
|
||||
if (results.multi_handedness[i].classification[0].label)=="Left":
|
||||
Left[0] = 1
|
||||
wrist = hand_landmarks.landmark[0]
|
||||
index_finger_mcp = hand_landmarks.landmark[5]
|
||||
index_finger_tip = hand_landmarks.landmark[8]
|
||||
middle_finger_mcp = hand_landmarks.landmark[9]
|
||||
middle_finger_tip = hand_landmarks.landmark[12]
|
||||
ring_finger_mcp = hand_landmarks.landmark[13]
|
||||
ring_finger_tip = hand_landmarks.landmark[16]
|
||||
pinky_mcp = hand_landmarks.landmark[17]
|
||||
pinky_tip = hand_landmarks.landmark[20]
|
||||
|
||||
one_x = index_finger_mcp.x - pinky_mcp.x
|
||||
one_y = index_finger_mcp.y - pinky_mcp.y
|
||||
|
||||
if (math.fabs(one_x) > math.fabs(one_y)):
|
||||
Left[3]=1
|
||||
if (one_x > 0):
|
||||
Left[1]=1
|
||||
tow = index_finger_mcp.y-wrist.y
|
||||
if (tow < 0): Left[2]=1
|
||||
length = math.fabs(middle_finger_mcp.y-wrist.y) / 3
|
||||
if (math.fabs(index_finger_tip.y-index_finger_mcp.y) > length): Left[5]=1
|
||||
if (math.fabs(middle_finger_tip.y-middle_finger_mcp.y) > length): Left[6]=1
|
||||
if (math.fabs(ring_finger_tip.y-ring_finger_mcp.y) > length): Left[7]=1
|
||||
if (math.fabs(pinky_tip.y-pinky_mcp.y) > length): Left[8]=1
|
||||
else:
|
||||
length = math.fabs(one_y) / 3
|
||||
if (math.fabs(index_finger_mcp.x-index_finger_tip.x) > length): Left[5]=1
|
||||
if (math.fabs(middle_finger_mcp.x-middle_finger_tip.x) > length): Left[6]=1
|
||||
if (math.fabs(ring_finger_mcp.x-ring_finger_tip.x) > length): Left[7]=1
|
||||
if (math.fabs(pinky_mcp.x-pinky_tip.x) > length): Left[8]=1
|
||||
else:
|
||||
Right[0] = 1
|
||||
wrist = hand_landmarks.landmark[0]
|
||||
index_finger_mcp = hand_landmarks.landmark[5]
|
||||
|
||||
index_finger_tip = hand_landmarks.landmark[8]
|
||||
|
||||
middle_finger_mcp = hand_landmarks.landmark[9]
|
||||
|
||||
middle_finger_tip = hand_landmarks.landmark[12]
|
||||
|
||||
ring_finger_mcp = hand_landmarks.landmark[13]
|
||||
|
||||
ring_finger_tip = hand_landmarks.landmark[16]
|
||||
|
||||
pinky_mcp = hand_landmarks.landmark[17]
|
||||
|
||||
pinky_tip = hand_landmarks.landmark[20]
|
||||
|
||||
one_x = index_finger_mcp.x - pinky_mcp.x
|
||||
one_y = index_finger_mcp.y - pinky_mcp.y
|
||||
if (math.fabs(one_x) > math.fabs(one_y)): Right[3]=1
|
||||
if (one_x > 0): Right[1]=1
|
||||
tow = index_finger_mcp.y-wrist.y
|
||||
if (tow < 0): Right[2]=1
|
||||
length = math.fabs(middle_finger_mcp.y-wrist.y) / 3
|
||||
if (math.fabs(index_finger_tip.y-index_finger_mcp.y) > length): Right[5]=1
|
||||
if (math.fabs(middle_finger_tip.y-middle_finger_mcp.y) > length): Right[6]=1
|
||||
if (math.fabs(ring_finger_tip.y-ring_finger_mcp.y) > length): Right[7]=1
|
||||
if (math.fabs(pinky_tip.y-pinky_mcp.y) > length): Right[8]=1
|
||||
else:
|
||||
length = math.fabs(one_y) / 2
|
||||
if (math.fabs(index_finger_mcp.x-index_finger_tip.x) > length): Right[5]=1
|
||||
if (math.fabs(middle_finger_mcp.x-middle_finger_tip.x) > length): Right[6]=1
|
||||
if (math.fabs(ring_finger_mcp.x-ring_finger_tip.x) > length): Right[7]=1
|
||||
if (math.fabs(pinky_mcp.x-pinky_tip.x) > length): Right[8]=1
|
||||
mp_drawing.draw_landmarks(
|
||||
image,
|
||||
hand_landmarks,
|
||||
mp_hands.HAND_CONNECTIONS,
|
||||
mp_drawing_styles.get_default_hand_landmarks_style(),
|
||||
mp_drawing_styles.get_default_hand_connections_style())
|
||||
# Flip the image horizontally for a selfie-view display.
|
||||
cv2.imshow('MediaPipe Hands', cv2.flip(image, 1))
|
||||
ctl = cv2.waitKey(5)
|
||||
if ctl == ord('q'):
|
||||
break
|
||||
elif ctl==ord('s'):
|
||||
print(Left,Right)
|
||||
# strs = "INSERT INTO XYZ VALUES ("
|
||||
# word = "\'"+input()+"\',"
|
||||
# strs+=word
|
||||
# print(aa)
|
||||
# print(len(wt))
|
||||
# for i in wt:
|
||||
# strs+=str(i)[:7]+","
|
||||
# strs = strs[:-1] + ")"
|
||||
cap.release()
|
||||
|
||||
|
||||
print ("数据插入成功")
|
||||
@ -0,0 +1,78 @@
|
||||
import sqlite3
|
||||
|
||||
conn = sqlite3.connect('info.db')
|
||||
c = conn.cursor()
|
||||
print ("数据库打开成功")
|
||||
strs = "INSERT INTO XYZ VALUES ("
|
||||
|
||||
|
||||
import cv2
|
||||
import mediapipe as mp
|
||||
|
||||
mp_drawing = mp.solutions.drawing_utils
|
||||
mp_drawing_styles = mp.solutions.drawing_styles
|
||||
mp_hands = mp.solutions.hands
|
||||
|
||||
num = 0
|
||||
# For webcam input:
|
||||
cap = cv2.VideoCapture(0)
|
||||
|
||||
with mp_hands.Hands(
|
||||
model_complexity=0,
|
||||
max_num_hands=2,
|
||||
min_detection_confidence=0.5,
|
||||
min_tracking_confidence=0.5) as hands:
|
||||
while cap.isOpened():
|
||||
success, image = cap.read()
|
||||
if not success:
|
||||
print("Ignoring empty camera frame.")
|
||||
# If loading a video, use 'break' instead of 'continue'.
|
||||
continue
|
||||
|
||||
# To improve performance, optionally mark the image as not writeable to
|
||||
# pass by reference.
|
||||
image.flags.writeable = False
|
||||
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
|
||||
results = hands.process(image)
|
||||
|
||||
# Draw the hand annotations on the image.
|
||||
image.flags.writeable = True
|
||||
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
|
||||
aa=0
|
||||
if results.multi_hand_landmarks:
|
||||
wt = []
|
||||
for i in range(42 * 3):
|
||||
wt.append(0)
|
||||
for i,hand_landmarks in enumerate(results.multi_hand_landmarks):
|
||||
if (results.multi_handedness[i].classification[0].label)=="Left":
|
||||
num = 0
|
||||
else:
|
||||
num =1
|
||||
for j,handlms in enumerate(hand_landmarks.landmark):
|
||||
wt[num*21*3+j*3],wt[num*21*3+j*3+1],wt[num*21*3+j*3+2] = handlms.x, handlms.y, handlms.z
|
||||
mp_drawing.draw_landmarks(
|
||||
image,
|
||||
hand_landmarks,
|
||||
mp_hands.HAND_CONNECTIONS,
|
||||
mp_drawing_styles.get_default_hand_landmarks_style(),
|
||||
mp_drawing_styles.get_default_hand_connections_style())
|
||||
# Flip the image horizontally for a selfie-view display.
|
||||
cv2.imshow('MediaPipe Hands', cv2.flip(image, 1))
|
||||
ctl = cv2.waitKey(5)
|
||||
if ctl == ord('q'):
|
||||
break
|
||||
elif ctl==ord('s'):
|
||||
strs = "INSERT INTO XYZ VALUES ("
|
||||
word = "\'"+input()+"\',"
|
||||
strs+=word
|
||||
print(aa)
|
||||
for i in wt:
|
||||
strs+=str(i)[:7]+","
|
||||
strs = strs[:-1] + ");"
|
||||
print(strs)
|
||||
c.execute(strs)
|
||||
cap.release()
|
||||
conn.commit()
|
||||
|
||||
print ("数据插入成功")
|
||||
conn.close()
|
||||
@ -0,0 +1,20 @@
|
||||
import sqlite3
|
||||
|
||||
conn = sqlite3.connect('info.db')
|
||||
c = conn.cursor()
|
||||
print ("数据库打开成功")
|
||||
def insert_xyz():
|
||||
strs = "INSERT INTO XYZ VALUES ('这',"
|
||||
for i in range(21*4):
|
||||
strs+=str(i+0.01)+","
|
||||
strs=strs[:-1]+");"
|
||||
c.execute(strs)
|
||||
|
||||
def insert_trans():
|
||||
strs = "INSERT INTO TRANSLATION VALUES('好','100000000|000000000');"
|
||||
print(strs)
|
||||
c.execute(strs)
|
||||
|
||||
conn.commit()
|
||||
print ("数据插入成功")
|
||||
conn.close()
|
||||
@ -0,0 +1,15 @@
|
||||
import sqlite3
|
||||
|
||||
conn = sqlite3.connect('info.db')
|
||||
print ("数据库打开成功")
|
||||
c = conn.cursor()
|
||||
strs = "CREATE TABLE TRANSLATION(WORD TEXT PRIMARY KEY NOT NULL,NUM TEXT);"
|
||||
# strnum = "CREATE TABLE XYZ(WORD TEXT PRIMARY KEY NOT NULL,"
|
||||
# for i in range(21*2*2):
|
||||
# strnum+="fig"+str(i).zfill(2)+" FLOAT NOT NULL,"
|
||||
# strnum=strnum[:-1]+");"
|
||||
print(strs)
|
||||
c.execute(strs)
|
||||
print ("数据表创建成功")
|
||||
conn.commit()
|
||||
conn.close()
|
||||
@ -0,0 +1,32 @@
|
||||
import tensorflow as tf
|
||||
import csv
|
||||
new_model = tf.keras.models.load_model('model')
|
||||
|
||||
# Check its architecture
|
||||
new_model.summary()
|
||||
|
||||
train_label = []
|
||||
train_feature=[]
|
||||
with open("test.csv","r") as r:
|
||||
reader = csv.reader(r)
|
||||
for i in reader:
|
||||
train_label.append((eval(i.pop())))
|
||||
b = [eval(j) for j in i]
|
||||
train_feature.append(b)
|
||||
train_features = tf.constant(train_feature)
|
||||
train_labels = tf.constant(train_label)
|
||||
|
||||
test_accuracy = tf.keras.metrics.Accuracy()
|
||||
|
||||
branch = 32
|
||||
for num in range(len(train_features)//branch):
|
||||
# training=False is needed only if there are layers with different
|
||||
# behavior during training versus inference (e.g. Dropout).
|
||||
x = train_features[num*32:(num+1)*32]
|
||||
y = train_labels[num*32:(num+1)*32]
|
||||
logits = new_model(x, training=False)
|
||||
prediction = tf.argmax(logits, axis=1, output_type=tf.int32)
|
||||
print("真实值为",y,"预测值为",prediction)
|
||||
test_accuracy(prediction, y)
|
||||
|
||||
print("Test set accuracy: {:.3%}".format(test_accuracy.result()))
|
||||
@ -0,0 +1,76 @@
|
||||
import cv2
|
||||
import mediapipe as mp
|
||||
import csv
|
||||
import tensorflow as tf
|
||||
import csv
|
||||
new_model = tf.keras.models.load_model('model')
|
||||
|
||||
# Check its architecture
|
||||
new_model.summary()
|
||||
|
||||
def test(inputs):
|
||||
# training=False is needed only if there are layers with different
|
||||
# behavior during training versus inference (e.g. Dropout).
|
||||
x = tf.constant([inputs])
|
||||
logits = new_model(x, training=False)
|
||||
prediction = tf.argmax(logits, axis=1, output_type=tf.int32)
|
||||
print(prediction)
|
||||
mp_drawing = mp.solutions.drawing_utils
|
||||
mp_drawing_styles = mp.solutions.drawing_styles
|
||||
mp_hands = mp.solutions.hands
|
||||
|
||||
ls = []
|
||||
for i in range(42*3*2):
|
||||
ls.append(0)
|
||||
wt = ls
|
||||
num = 0
|
||||
# For webcam input:
|
||||
cap = cv2.VideoCapture(0)
|
||||
|
||||
|
||||
with open("test.csv","a") as csvfile:
|
||||
writer = csv.writer(csvfile)
|
||||
|
||||
with mp_hands.Hands(
|
||||
model_complexity=0,
|
||||
max_num_hands=2,
|
||||
min_detection_confidence=0.5,
|
||||
min_tracking_confidence=0.5) as hands:
|
||||
while cap.isOpened():
|
||||
success, image = cap.read()
|
||||
if not success:
|
||||
print("Ignoring empty camera frame.")
|
||||
# If loading a video, use 'break' instead of 'continue'.
|
||||
continue
|
||||
|
||||
# To improve performance, optionally mark the image as not writeable to
|
||||
# pass by reference.
|
||||
image.flags.writeable = False
|
||||
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
|
||||
results = hands.process(image)
|
||||
|
||||
# Draw the hand annotations on the image.
|
||||
image.flags.writeable = True
|
||||
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
|
||||
|
||||
if results.multi_hand_landmarks:
|
||||
for i,hand_landmarks in enumerate(results.multi_hand_landmarks):
|
||||
if (results.multi_handedness[i].classification[0].label)=="Left":
|
||||
num = 0
|
||||
else:
|
||||
num =1
|
||||
for j,handlms in enumerate(hand_landmarks.landmark):
|
||||
wt[num*21*3+j],wt[num*21*3+j+1],wt[num*21*3+j+2] = handlms.x, handlms.y, handlms.z
|
||||
mp_drawing.draw_landmarks(
|
||||
image,
|
||||
hand_landmarks,
|
||||
mp_hands.HAND_CONNECTIONS,
|
||||
mp_drawing_styles.get_default_hand_landmarks_style(),
|
||||
mp_drawing_styles.get_default_hand_connections_style())
|
||||
# Flip the image horizontally for a selfie-view display.
|
||||
test(wt)
|
||||
cv2.imshow('MediaPipe Hands', cv2.flip(image, 1))
|
||||
ctl = cv2.waitKey(5)
|
||||
if ctl == ord('q'):
|
||||
break
|
||||
cap.release()
|
||||
@ -0,0 +1,607 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/framework/calculator_profile.proto
|
||||
|
||||
import sys
|
||||
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
|
||||
from google.protobuf import descriptor as _descriptor
|
||||
from google.protobuf import message as _message
|
||||
from google.protobuf import reflection as _reflection
|
||||
from google.protobuf import symbol_database as _symbol_database
|
||||
# @@protoc_insertion_point(imports)
|
||||
|
||||
_sym_db = _symbol_database.Default()
|
||||
|
||||
|
||||
from mediapipe.framework import calculator_pb2 as mediapipe_dot_framework_dot_calculator__pb2
|
||||
try:
|
||||
mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe_dot_framework_dot_calculator__options__pb2
|
||||
except AttributeError:
|
||||
mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe.framework.calculator_options_pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/framework/calculator_profile.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=_b('\n\032com.google.mediapipe.protoB\026CalculatorProfileProto'),
|
||||
serialized_pb=_b('\n,mediapipe/framework/calculator_profile.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"o\n\rTimeHistogram\x12\x10\n\x05total\x18\x01 \x01(\x03:\x01\x30\x12#\n\x12interval_size_usec\x18\x02 \x01(\x03:\x07\x31\x30\x30\x30\x30\x30\x30\x12\x18\n\rnum_intervals\x18\x03 \x01(\x03:\x01\x31\x12\r\n\x05\x63ount\x18\x04 \x03(\x03\"b\n\rStreamProfile\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x18\n\tback_edge\x18\x02 \x01(\x08:\x05\x66\x61lse\x12)\n\x07latency\x18\x03 \x01(\x0b\x32\x18.mediapipe.TimeHistogram\"\xb3\x02\n\x11\x43\x61lculatorProfile\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x17\n\x0copen_runtime\x18\x02 \x01(\x03:\x01\x30\x12\x18\n\rclose_runtime\x18\x03 \x01(\x03:\x01\x30\x12\x31\n\x0fprocess_runtime\x18\x04 \x01(\x0b\x32\x18.mediapipe.TimeHistogram\x12\x37\n\x15process_input_latency\x18\x05 \x01(\x0b\x32\x18.mediapipe.TimeHistogram\x12\x38\n\x16process_output_latency\x18\x06 \x01(\x0b\x32\x18.mediapipe.TimeHistogram\x12\x37\n\x15input_stream_profiles\x18\x07 \x03(\x0b\x32\x18.mediapipe.StreamProfile\"\xe9\x06\n\nGraphTrace\x12\x11\n\tbase_time\x18\x01 \x01(\x03\x12\x16\n\x0e\x62\x61se_timestamp\x18\x02 \x01(\x03\x12\x17\n\x0f\x63\x61lculator_name\x18\x03 \x03(\t\x12\x13\n\x0bstream_name\x18\x04 \x03(\t\x12?\n\x10\x63\x61lculator_trace\x18\x05 \x03(\x0b\x32%.mediapipe.GraphTrace.CalculatorTrace\x1a\x8e\x01\n\x0bStreamTrace\x12\x12\n\nstart_time\x18\x01 \x01(\x03\x12\x13\n\x0b\x66inish_time\x18\x02 \x01(\x03\x12\x18\n\x10packet_timestamp\x18\x03 \x01(\x03\x12\x11\n\tstream_id\x18\x04 \x01(\x05\x12\x15\n\tpacket_id\x18\x05 \x01(\x03\x42\x02\x18\x01\x12\x12\n\nevent_data\x18\x06 \x01(\x03\x1a\x9d\x02\n\x0f\x43\x61lculatorTrace\x12\x0f\n\x07node_id\x18\x01 \x01(\x05\x12\x17\n\x0finput_timestamp\x18\x02 \x01(\x03\x12\x33\n\nevent_type\x18\x03 \x01(\x0e\x32\x1f.mediapipe.GraphTrace.EventType\x12\x12\n\nstart_time\x18\x04 \x01(\x03\x12\x13\n\x0b\x66inish_time\x18\x05 \x01(\x03\x12\x36\n\x0binput_trace\x18\x06 \x03(\x0b\x32!.mediapipe.GraphTrace.StreamTrace\x12\x37\n\x0coutput_trace\x18\x07 \x03(\x0b\x32!.mediapipe.GraphTrace.StreamTrace\x12\x11\n\tthread_id\x18\x08 \x01(\x05\"\x8f\x02\n\tEventType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x08\n\x04OPEN\x10\x01\x12\x0b\n\x07PROCESS\x10\x02\x12\t\n\x05\x43LOSE\x10\x03\x12\r\n\tNOT_READY\x10\x04\x12\x15\n\x11READY_FOR_PROCESS\x10\x05\x12\x13\n\x0fREADY_FOR_CLOSE\x10\x06\x12\r\n\tTHROTTLED\x10\x07\x12\x0f\n\x0bUNTHROTTLED\x10\x08\x12\x11\n\rCPU_TASK_USER\x10\t\x12\x13\n\x0f\x43PU_TASK_SYSTEM\x10\n\x12\x0c\n\x08GPU_TASK\x10\x0b\x12\x0c\n\x08\x44SP_TASK\x10\x0c\x12\x0c\n\x08TPU_TASK\x10\r\x12\x13\n\x0fGPU_CALIBRATION\x10\x0e\x12\x11\n\rPACKET_QUEUED\x10\x0f\"\xa7\x01\n\x0cGraphProfile\x12*\n\x0bgraph_trace\x18\x01 \x03(\x0b\x32\x15.mediapipe.GraphTrace\x12\x39\n\x13\x63\x61lculator_profiles\x18\x02 \x03(\x0b\x32\x1c.mediapipe.CalculatorProfile\x12\x30\n\x06\x63onfig\x18\x03 \x01(\x0b\x32 .mediapipe.CalculatorGraphConfigB4\n\x1a\x63om.google.mediapipe.protoB\x16\x43\x61lculatorProfileProto')
|
||||
,
|
||||
dependencies=[mediapipe_dot_framework_dot_calculator__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
_GRAPHTRACE_EVENTTYPE = _descriptor.EnumDescriptor(
|
||||
name='EventType',
|
||||
full_name='mediapipe.GraphTrace.EventType',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='UNKNOWN', index=0, number=0,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='OPEN', index=1, number=1,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='PROCESS', index=2, number=2,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='CLOSE', index=3, number=3,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='NOT_READY', index=4, number=4,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='READY_FOR_PROCESS', index=5, number=5,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='READY_FOR_CLOSE', index=6, number=6,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='THROTTLED', index=7, number=7,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='UNTHROTTLED', index=8, number=8,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='CPU_TASK_USER', index=9, number=9,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='CPU_TASK_SYSTEM', index=10, number=10,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='GPU_TASK', index=11, number=11,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='DSP_TASK', index=12, number=12,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='TPU_TASK', index=13, number=13,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='GPU_CALIBRATION', index=14, number=14,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='PACKET_QUEUED', index=15, number=15,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=1223,
|
||||
serialized_end=1494,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_GRAPHTRACE_EVENTTYPE)
|
||||
|
||||
|
||||
_TIMEHISTOGRAM = _descriptor.Descriptor(
|
||||
name='TimeHistogram',
|
||||
full_name='mediapipe.TimeHistogram',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='total', full_name='mediapipe.TimeHistogram.total', index=0,
|
||||
number=1, type=3, cpp_type=2, label=1,
|
||||
has_default_value=True, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='interval_size_usec', full_name='mediapipe.TimeHistogram.interval_size_usec', index=1,
|
||||
number=2, type=3, cpp_type=2, label=1,
|
||||
has_default_value=True, default_value=1000000,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='num_intervals', full_name='mediapipe.TimeHistogram.num_intervals', index=2,
|
||||
number=3, type=3, cpp_type=2, label=1,
|
||||
has_default_value=True, default_value=1,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='count', full_name='mediapipe.TimeHistogram.count', index=3,
|
||||
number=4, type=3, cpp_type=2, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=97,
|
||||
serialized_end=208,
|
||||
)
|
||||
|
||||
|
||||
_STREAMPROFILE = _descriptor.Descriptor(
|
||||
name='StreamProfile',
|
||||
full_name='mediapipe.StreamProfile',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='name', full_name='mediapipe.StreamProfile.name', index=0,
|
||||
number=1, type=9, cpp_type=9, label=1,
|
||||
has_default_value=False, default_value=_b("").decode('utf-8'),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='back_edge', full_name='mediapipe.StreamProfile.back_edge', index=1,
|
||||
number=2, type=8, cpp_type=7, label=1,
|
||||
has_default_value=True, default_value=False,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='latency', full_name='mediapipe.StreamProfile.latency', index=2,
|
||||
number=3, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=210,
|
||||
serialized_end=308,
|
||||
)
|
||||
|
||||
|
||||
_CALCULATORPROFILE = _descriptor.Descriptor(
|
||||
name='CalculatorProfile',
|
||||
full_name='mediapipe.CalculatorProfile',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='name', full_name='mediapipe.CalculatorProfile.name', index=0,
|
||||
number=1, type=9, cpp_type=9, label=1,
|
||||
has_default_value=False, default_value=_b("").decode('utf-8'),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='open_runtime', full_name='mediapipe.CalculatorProfile.open_runtime', index=1,
|
||||
number=2, type=3, cpp_type=2, label=1,
|
||||
has_default_value=True, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='close_runtime', full_name='mediapipe.CalculatorProfile.close_runtime', index=2,
|
||||
number=3, type=3, cpp_type=2, label=1,
|
||||
has_default_value=True, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='process_runtime', full_name='mediapipe.CalculatorProfile.process_runtime', index=3,
|
||||
number=4, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='process_input_latency', full_name='mediapipe.CalculatorProfile.process_input_latency', index=4,
|
||||
number=5, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='process_output_latency', full_name='mediapipe.CalculatorProfile.process_output_latency', index=5,
|
||||
number=6, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='input_stream_profiles', full_name='mediapipe.CalculatorProfile.input_stream_profiles', index=6,
|
||||
number=7, type=11, cpp_type=10, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=311,
|
||||
serialized_end=618,
|
||||
)
|
||||
|
||||
|
||||
_GRAPHTRACE_STREAMTRACE = _descriptor.Descriptor(
|
||||
name='StreamTrace',
|
||||
full_name='mediapipe.GraphTrace.StreamTrace',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='start_time', full_name='mediapipe.GraphTrace.StreamTrace.start_time', index=0,
|
||||
number=1, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='finish_time', full_name='mediapipe.GraphTrace.StreamTrace.finish_time', index=1,
|
||||
number=2, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='packet_timestamp', full_name='mediapipe.GraphTrace.StreamTrace.packet_timestamp', index=2,
|
||||
number=3, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='stream_id', full_name='mediapipe.GraphTrace.StreamTrace.stream_id', index=3,
|
||||
number=4, type=5, cpp_type=1, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='packet_id', full_name='mediapipe.GraphTrace.StreamTrace.packet_id', index=4,
|
||||
number=5, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=_b('\030\001'), file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='event_data', full_name='mediapipe.GraphTrace.StreamTrace.event_data', index=5,
|
||||
number=6, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=790,
|
||||
serialized_end=932,
|
||||
)
|
||||
|
||||
_GRAPHTRACE_CALCULATORTRACE = _descriptor.Descriptor(
|
||||
name='CalculatorTrace',
|
||||
full_name='mediapipe.GraphTrace.CalculatorTrace',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='node_id', full_name='mediapipe.GraphTrace.CalculatorTrace.node_id', index=0,
|
||||
number=1, type=5, cpp_type=1, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='input_timestamp', full_name='mediapipe.GraphTrace.CalculatorTrace.input_timestamp', index=1,
|
||||
number=2, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='event_type', full_name='mediapipe.GraphTrace.CalculatorTrace.event_type', index=2,
|
||||
number=3, type=14, cpp_type=8, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='start_time', full_name='mediapipe.GraphTrace.CalculatorTrace.start_time', index=3,
|
||||
number=4, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='finish_time', full_name='mediapipe.GraphTrace.CalculatorTrace.finish_time', index=4,
|
||||
number=5, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='input_trace', full_name='mediapipe.GraphTrace.CalculatorTrace.input_trace', index=5,
|
||||
number=6, type=11, cpp_type=10, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='output_trace', full_name='mediapipe.GraphTrace.CalculatorTrace.output_trace', index=6,
|
||||
number=7, type=11, cpp_type=10, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='thread_id', full_name='mediapipe.GraphTrace.CalculatorTrace.thread_id', index=7,
|
||||
number=8, type=5, cpp_type=1, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=935,
|
||||
serialized_end=1220,
|
||||
)
|
||||
|
||||
_GRAPHTRACE = _descriptor.Descriptor(
|
||||
name='GraphTrace',
|
||||
full_name='mediapipe.GraphTrace',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='base_time', full_name='mediapipe.GraphTrace.base_time', index=0,
|
||||
number=1, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='base_timestamp', full_name='mediapipe.GraphTrace.base_timestamp', index=1,
|
||||
number=2, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='calculator_name', full_name='mediapipe.GraphTrace.calculator_name', index=2,
|
||||
number=3, type=9, cpp_type=9, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='stream_name', full_name='mediapipe.GraphTrace.stream_name', index=3,
|
||||
number=4, type=9, cpp_type=9, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='calculator_trace', full_name='mediapipe.GraphTrace.calculator_trace', index=4,
|
||||
number=5, type=11, cpp_type=10, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[_GRAPHTRACE_STREAMTRACE, _GRAPHTRACE_CALCULATORTRACE, ],
|
||||
enum_types=[
|
||||
_GRAPHTRACE_EVENTTYPE,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=621,
|
||||
serialized_end=1494,
|
||||
)
|
||||
|
||||
|
||||
_GRAPHPROFILE = _descriptor.Descriptor(
|
||||
name='GraphProfile',
|
||||
full_name='mediapipe.GraphProfile',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='graph_trace', full_name='mediapipe.GraphProfile.graph_trace', index=0,
|
||||
number=1, type=11, cpp_type=10, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='calculator_profiles', full_name='mediapipe.GraphProfile.calculator_profiles', index=1,
|
||||
number=2, type=11, cpp_type=10, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='config', full_name='mediapipe.GraphProfile.config', index=2,
|
||||
number=3, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1497,
|
||||
serialized_end=1664,
|
||||
)
|
||||
|
||||
_STREAMPROFILE.fields_by_name['latency'].message_type = _TIMEHISTOGRAM
|
||||
_CALCULATORPROFILE.fields_by_name['process_runtime'].message_type = _TIMEHISTOGRAM
|
||||
_CALCULATORPROFILE.fields_by_name['process_input_latency'].message_type = _TIMEHISTOGRAM
|
||||
_CALCULATORPROFILE.fields_by_name['process_output_latency'].message_type = _TIMEHISTOGRAM
|
||||
_CALCULATORPROFILE.fields_by_name['input_stream_profiles'].message_type = _STREAMPROFILE
|
||||
_GRAPHTRACE_STREAMTRACE.containing_type = _GRAPHTRACE
|
||||
_GRAPHTRACE_CALCULATORTRACE.fields_by_name['event_type'].enum_type = _GRAPHTRACE_EVENTTYPE
|
||||
_GRAPHTRACE_CALCULATORTRACE.fields_by_name['input_trace'].message_type = _GRAPHTRACE_STREAMTRACE
|
||||
_GRAPHTRACE_CALCULATORTRACE.fields_by_name['output_trace'].message_type = _GRAPHTRACE_STREAMTRACE
|
||||
_GRAPHTRACE_CALCULATORTRACE.containing_type = _GRAPHTRACE
|
||||
_GRAPHTRACE.fields_by_name['calculator_trace'].message_type = _GRAPHTRACE_CALCULATORTRACE
|
||||
_GRAPHTRACE_EVENTTYPE.containing_type = _GRAPHTRACE
|
||||
_GRAPHPROFILE.fields_by_name['graph_trace'].message_type = _GRAPHTRACE
|
||||
_GRAPHPROFILE.fields_by_name['calculator_profiles'].message_type = _CALCULATORPROFILE
|
||||
_GRAPHPROFILE.fields_by_name['config'].message_type = mediapipe_dot_framework_dot_calculator__pb2._CALCULATORGRAPHCONFIG
|
||||
DESCRIPTOR.message_types_by_name['TimeHistogram'] = _TIMEHISTOGRAM
|
||||
DESCRIPTOR.message_types_by_name['StreamProfile'] = _STREAMPROFILE
|
||||
DESCRIPTOR.message_types_by_name['CalculatorProfile'] = _CALCULATORPROFILE
|
||||
DESCRIPTOR.message_types_by_name['GraphTrace'] = _GRAPHTRACE
|
||||
DESCRIPTOR.message_types_by_name['GraphProfile'] = _GRAPHPROFILE
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
TimeHistogram = _reflection.GeneratedProtocolMessageType('TimeHistogram', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TIMEHISTOGRAM,
|
||||
__module__ = 'mediapipe.framework.calculator_profile_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TimeHistogram)
|
||||
))
|
||||
_sym_db.RegisterMessage(TimeHistogram)
|
||||
|
||||
StreamProfile = _reflection.GeneratedProtocolMessageType('StreamProfile', (_message.Message,), dict(
|
||||
DESCRIPTOR = _STREAMPROFILE,
|
||||
__module__ = 'mediapipe.framework.calculator_profile_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.StreamProfile)
|
||||
))
|
||||
_sym_db.RegisterMessage(StreamProfile)
|
||||
|
||||
CalculatorProfile = _reflection.GeneratedProtocolMessageType('CalculatorProfile', (_message.Message,), dict(
|
||||
DESCRIPTOR = _CALCULATORPROFILE,
|
||||
__module__ = 'mediapipe.framework.calculator_profile_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.CalculatorProfile)
|
||||
))
|
||||
_sym_db.RegisterMessage(CalculatorProfile)
|
||||
|
||||
GraphTrace = _reflection.GeneratedProtocolMessageType('GraphTrace', (_message.Message,), dict(
|
||||
|
||||
StreamTrace = _reflection.GeneratedProtocolMessageType('StreamTrace', (_message.Message,), dict(
|
||||
DESCRIPTOR = _GRAPHTRACE_STREAMTRACE,
|
||||
__module__ = 'mediapipe.framework.calculator_profile_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.GraphTrace.StreamTrace)
|
||||
))
|
||||
,
|
||||
|
||||
CalculatorTrace = _reflection.GeneratedProtocolMessageType('CalculatorTrace', (_message.Message,), dict(
|
||||
DESCRIPTOR = _GRAPHTRACE_CALCULATORTRACE,
|
||||
__module__ = 'mediapipe.framework.calculator_profile_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.GraphTrace.CalculatorTrace)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _GRAPHTRACE,
|
||||
__module__ = 'mediapipe.framework.calculator_profile_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.GraphTrace)
|
||||
))
|
||||
_sym_db.RegisterMessage(GraphTrace)
|
||||
_sym_db.RegisterMessage(GraphTrace.StreamTrace)
|
||||
_sym_db.RegisterMessage(GraphTrace.CalculatorTrace)
|
||||
|
||||
GraphProfile = _reflection.GeneratedProtocolMessageType('GraphProfile', (_message.Message,), dict(
|
||||
DESCRIPTOR = _GRAPHPROFILE,
|
||||
__module__ = 'mediapipe.framework.calculator_profile_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.GraphProfile)
|
||||
))
|
||||
_sym_db.RegisterMessage(GraphProfile)
|
||||
|
||||
|
||||
DESCRIPTOR._options = None
|
||||
_GRAPHTRACE_STREAMTRACE.fields_by_name['packet_id']._options = None
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@ -0,0 +1,64 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/framework/mediapipe_options.proto
|
||||
|
||||
import sys
|
||||
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
|
||||
from google.protobuf import descriptor as _descriptor
|
||||
from google.protobuf import message as _message
|
||||
from google.protobuf import reflection as _reflection
|
||||
from google.protobuf import symbol_database as _symbol_database
|
||||
# @@protoc_insertion_point(imports)
|
||||
|
||||
_sym_db = _symbol_database.Default()
|
||||
|
||||
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/framework/mediapipe_options.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=_b('\n\032com.google.mediapipe.protoB\025MediaPipeOptionsProto'),
|
||||
serialized_pb=_b('\n+mediapipe/framework/mediapipe_options.proto\x12\tmediapipe\"\x1e\n\x10MediaPipeOptions*\n\x08\xa0\x9c\x01\x10\x80\x80\x80\x80\x02\x42\x33\n\x1a\x63om.google.mediapipe.protoB\x15MediaPipeOptionsProto')
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
_MEDIAPIPEOPTIONS = _descriptor.Descriptor(
|
||||
name='MediaPipeOptions',
|
||||
full_name='mediapipe.MediaPipeOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=True,
|
||||
syntax='proto2',
|
||||
extension_ranges=[(20000, 536870912), ],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=58,
|
||||
serialized_end=88,
|
||||
)
|
||||
|
||||
DESCRIPTOR.message_types_by_name['MediaPipeOptions'] = _MEDIAPIPEOPTIONS
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
MediaPipeOptions = _reflection.GeneratedProtocolMessageType('MediaPipeOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _MEDIAPIPEOPTIONS,
|
||||
__module__ = 'mediapipe.framework.mediapipe_options_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MediaPipeOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(MediaPipeOptions)
|
||||
|
||||
|
||||
DESCRIPTOR._options = None
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@ -0,0 +1,165 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/framework/packet_factory.proto
|
||||
|
||||
import sys
|
||||
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
|
||||
from google.protobuf import descriptor as _descriptor
|
||||
from google.protobuf import message as _message
|
||||
from google.protobuf import reflection as _reflection
|
||||
from google.protobuf import symbol_database as _symbol_database
|
||||
# @@protoc_insertion_point(imports)
|
||||
|
||||
_sym_db = _symbol_database.Default()
|
||||
|
||||
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/framework/packet_factory.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=_b('\n\032com.google.mediapipe.protoB\022PacketFactoryProto'),
|
||||
serialized_pb=_b('\n(mediapipe/framework/packet_factory.proto\x12\tmediapipe\"\"\n\x14PacketFactoryOptions*\n\x08\xa0\x9c\x01\x10\x80\x80\x80\x80\x02\"\x95\x01\n\x13PacketFactoryConfig\x12\x16\n\x0epacket_factory\x18\x01 \x01(\t\x12\x1a\n\x12output_side_packet\x18\x02 \x01(\t\x12\x18\n\x0f\x65xternal_output\x18\xea\x07 \x01(\t\x12\x30\n\x07options\x18\x03 \x01(\x0b\x32\x1f.mediapipe.PacketFactoryOptions\"E\n\x13PacketManagerConfig\x12.\n\x06packet\x18\x01 \x03(\x0b\x32\x1e.mediapipe.PacketFactoryConfigB0\n\x1a\x63om.google.mediapipe.protoB\x12PacketFactoryProto')
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
_PACKETFACTORYOPTIONS = _descriptor.Descriptor(
|
||||
name='PacketFactoryOptions',
|
||||
full_name='mediapipe.PacketFactoryOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=True,
|
||||
syntax='proto2',
|
||||
extension_ranges=[(20000, 536870912), ],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=55,
|
||||
serialized_end=89,
|
||||
)
|
||||
|
||||
|
||||
_PACKETFACTORYCONFIG = _descriptor.Descriptor(
|
||||
name='PacketFactoryConfig',
|
||||
full_name='mediapipe.PacketFactoryConfig',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='packet_factory', full_name='mediapipe.PacketFactoryConfig.packet_factory', index=0,
|
||||
number=1, type=9, cpp_type=9, label=1,
|
||||
has_default_value=False, default_value=_b("").decode('utf-8'),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='output_side_packet', full_name='mediapipe.PacketFactoryConfig.output_side_packet', index=1,
|
||||
number=2, type=9, cpp_type=9, label=1,
|
||||
has_default_value=False, default_value=_b("").decode('utf-8'),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='external_output', full_name='mediapipe.PacketFactoryConfig.external_output', index=2,
|
||||
number=1002, type=9, cpp_type=9, label=1,
|
||||
has_default_value=False, default_value=_b("").decode('utf-8'),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='options', full_name='mediapipe.PacketFactoryConfig.options', index=3,
|
||||
number=3, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=92,
|
||||
serialized_end=241,
|
||||
)
|
||||
|
||||
|
||||
_PACKETMANAGERCONFIG = _descriptor.Descriptor(
|
||||
name='PacketManagerConfig',
|
||||
full_name='mediapipe.PacketManagerConfig',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='packet', full_name='mediapipe.PacketManagerConfig.packet', index=0,
|
||||
number=1, type=11, cpp_type=10, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=243,
|
||||
serialized_end=312,
|
||||
)
|
||||
|
||||
_PACKETFACTORYCONFIG.fields_by_name['options'].message_type = _PACKETFACTORYOPTIONS
|
||||
_PACKETMANAGERCONFIG.fields_by_name['packet'].message_type = _PACKETFACTORYCONFIG
|
||||
DESCRIPTOR.message_types_by_name['PacketFactoryOptions'] = _PACKETFACTORYOPTIONS
|
||||
DESCRIPTOR.message_types_by_name['PacketFactoryConfig'] = _PACKETFACTORYCONFIG
|
||||
DESCRIPTOR.message_types_by_name['PacketManagerConfig'] = _PACKETMANAGERCONFIG
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
PacketFactoryOptions = _reflection.GeneratedProtocolMessageType('PacketFactoryOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _PACKETFACTORYOPTIONS,
|
||||
__module__ = 'mediapipe.framework.packet_factory_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.PacketFactoryOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(PacketFactoryOptions)
|
||||
|
||||
PacketFactoryConfig = _reflection.GeneratedProtocolMessageType('PacketFactoryConfig', (_message.Message,), dict(
|
||||
DESCRIPTOR = _PACKETFACTORYCONFIG,
|
||||
__module__ = 'mediapipe.framework.packet_factory_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.PacketFactoryConfig)
|
||||
))
|
||||
_sym_db.RegisterMessage(PacketFactoryConfig)
|
||||
|
||||
PacketManagerConfig = _reflection.GeneratedProtocolMessageType('PacketManagerConfig', (_message.Message,), dict(
|
||||
DESCRIPTOR = _PACKETMANAGERCONFIG,
|
||||
__module__ = 'mediapipe.framework.packet_factory_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.PacketManagerConfig)
|
||||
))
|
||||
_sym_db.RegisterMessage(PacketManagerConfig)
|
||||
|
||||
|
||||
DESCRIPTOR._options = None
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@ -0,0 +1,146 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/framework/packet_generator.proto
|
||||
|
||||
import sys
|
||||
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
|
||||
from google.protobuf import descriptor as _descriptor
|
||||
from google.protobuf import message as _message
|
||||
from google.protobuf import reflection as _reflection
|
||||
from google.protobuf import symbol_database as _symbol_database
|
||||
# @@protoc_insertion_point(imports)
|
||||
|
||||
_sym_db = _symbol_database.Default()
|
||||
|
||||
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/framework/packet_generator.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=_b('\n\032com.google.mediapipe.protoB\024PacketGeneratorProto'),
|
||||
serialized_pb=_b('\n*mediapipe/framework/packet_generator.proto\x12\tmediapipe\"@\n\x16PacketGeneratorOptions\x12\x1a\n\x0cmerge_fields\x18\x01 \x01(\x08:\x04true*\n\x08\xa0\x9c\x01\x10\x80\x80\x80\x80\x02\"\xcf\x01\n\x15PacketGeneratorConfig\x12\x18\n\x10packet_generator\x18\x01 \x01(\t\x12\x19\n\x11input_side_packet\x18\x02 \x03(\t\x12\x17\n\x0e\x65xternal_input\x18\xea\x07 \x03(\t\x12\x1a\n\x12output_side_packet\x18\x03 \x03(\t\x12\x18\n\x0f\x65xternal_output\x18\xeb\x07 \x03(\t\x12\x32\n\x07options\x18\x04 \x01(\x0b\x32!.mediapipe.PacketGeneratorOptionsB2\n\x1a\x63om.google.mediapipe.protoB\x14PacketGeneratorProto')
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
_PACKETGENERATOROPTIONS = _descriptor.Descriptor(
|
||||
name='PacketGeneratorOptions',
|
||||
full_name='mediapipe.PacketGeneratorOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='merge_fields', full_name='mediapipe.PacketGeneratorOptions.merge_fields', index=0,
|
||||
number=1, type=8, cpp_type=7, label=1,
|
||||
has_default_value=True, default_value=True,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=True,
|
||||
syntax='proto2',
|
||||
extension_ranges=[(20000, 536870912), ],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=57,
|
||||
serialized_end=121,
|
||||
)
|
||||
|
||||
|
||||
_PACKETGENERATORCONFIG = _descriptor.Descriptor(
|
||||
name='PacketGeneratorConfig',
|
||||
full_name='mediapipe.PacketGeneratorConfig',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='packet_generator', full_name='mediapipe.PacketGeneratorConfig.packet_generator', index=0,
|
||||
number=1, type=9, cpp_type=9, label=1,
|
||||
has_default_value=False, default_value=_b("").decode('utf-8'),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='input_side_packet', full_name='mediapipe.PacketGeneratorConfig.input_side_packet', index=1,
|
||||
number=2, type=9, cpp_type=9, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='external_input', full_name='mediapipe.PacketGeneratorConfig.external_input', index=2,
|
||||
number=1002, type=9, cpp_type=9, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='output_side_packet', full_name='mediapipe.PacketGeneratorConfig.output_side_packet', index=3,
|
||||
number=3, type=9, cpp_type=9, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='external_output', full_name='mediapipe.PacketGeneratorConfig.external_output', index=4,
|
||||
number=1003, type=9, cpp_type=9, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='options', full_name='mediapipe.PacketGeneratorConfig.options', index=5,
|
||||
number=4, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=124,
|
||||
serialized_end=331,
|
||||
)
|
||||
|
||||
_PACKETGENERATORCONFIG.fields_by_name['options'].message_type = _PACKETGENERATOROPTIONS
|
||||
DESCRIPTOR.message_types_by_name['PacketGeneratorOptions'] = _PACKETGENERATOROPTIONS
|
||||
DESCRIPTOR.message_types_by_name['PacketGeneratorConfig'] = _PACKETGENERATORCONFIG
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
PacketGeneratorOptions = _reflection.GeneratedProtocolMessageType('PacketGeneratorOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _PACKETGENERATOROPTIONS,
|
||||
__module__ = 'mediapipe.framework.packet_generator_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.PacketGeneratorOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(PacketGeneratorOptions)
|
||||
|
||||
PacketGeneratorConfig = _reflection.GeneratedProtocolMessageType('PacketGeneratorConfig', (_message.Message,), dict(
|
||||
DESCRIPTOR = _PACKETGENERATORCONFIG,
|
||||
__module__ = 'mediapipe.framework.packet_generator_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.PacketGeneratorConfig)
|
||||
))
|
||||
_sym_db.RegisterMessage(PacketGeneratorConfig)
|
||||
|
||||
|
||||
DESCRIPTOR._options = None
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@ -0,0 +1,95 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/framework/status_handler.proto
|
||||
|
||||
import sys
|
||||
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
|
||||
from google.protobuf import descriptor as _descriptor
|
||||
from google.protobuf import message as _message
|
||||
from google.protobuf import reflection as _reflection
|
||||
from google.protobuf import symbol_database as _symbol_database
|
||||
# @@protoc_insertion_point(imports)
|
||||
|
||||
_sym_db = _symbol_database.Default()
|
||||
|
||||
|
||||
from mediapipe.framework import mediapipe_options_pb2 as mediapipe_dot_framework_dot_mediapipe__options__pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/framework/status_handler.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=_b('\n\032com.google.mediapipe.protoB\022StatusHandlerProto'),
|
||||
serialized_pb=_b('\n(mediapipe/framework/status_handler.proto\x12\tmediapipe\x1a+mediapipe/framework/mediapipe_options.proto\"\x8f\x01\n\x13StatusHandlerConfig\x12\x16\n\x0estatus_handler\x18\x01 \x01(\t\x12\x19\n\x11input_side_packet\x18\x02 \x03(\t\x12\x17\n\x0e\x65xternal_input\x18\xea\x07 \x03(\t\x12,\n\x07options\x18\x03 \x01(\x0b\x32\x1b.mediapipe.MediaPipeOptionsB0\n\x1a\x63om.google.mediapipe.protoB\x12StatusHandlerProto')
|
||||
,
|
||||
dependencies=[mediapipe_dot_framework_dot_mediapipe__options__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
|
||||
_STATUSHANDLERCONFIG = _descriptor.Descriptor(
|
||||
name='StatusHandlerConfig',
|
||||
full_name='mediapipe.StatusHandlerConfig',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='status_handler', full_name='mediapipe.StatusHandlerConfig.status_handler', index=0,
|
||||
number=1, type=9, cpp_type=9, label=1,
|
||||
has_default_value=False, default_value=_b("").decode('utf-8'),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='input_side_packet', full_name='mediapipe.StatusHandlerConfig.input_side_packet', index=1,
|
||||
number=2, type=9, cpp_type=9, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='external_input', full_name='mediapipe.StatusHandlerConfig.external_input', index=2,
|
||||
number=1002, type=9, cpp_type=9, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='options', full_name='mediapipe.StatusHandlerConfig.options', index=3,
|
||||
number=3, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=101,
|
||||
serialized_end=244,
|
||||
)
|
||||
|
||||
_STATUSHANDLERCONFIG.fields_by_name['options'].message_type = mediapipe_dot_framework_dot_mediapipe__options__pb2._MEDIAPIPEOPTIONS
|
||||
DESCRIPTOR.message_types_by_name['StatusHandlerConfig'] = _STATUSHANDLERCONFIG
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
StatusHandlerConfig = _reflection.GeneratedProtocolMessageType('StatusHandlerConfig', (_message.Message,), dict(
|
||||
DESCRIPTOR = _STATUSHANDLERCONFIG,
|
||||
__module__ = 'mediapipe.framework.status_handler_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.StatusHandlerConfig)
|
||||
))
|
||||
_sym_db.RegisterMessage(StatusHandlerConfig)
|
||||
|
||||
|
||||
DESCRIPTOR._options = None
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@ -0,0 +1,135 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/framework/stream_handler.proto
|
||||
|
||||
import sys
|
||||
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
|
||||
from google.protobuf import descriptor as _descriptor
|
||||
from google.protobuf import message as _message
|
||||
from google.protobuf import reflection as _reflection
|
||||
from google.protobuf import symbol_database as _symbol_database
|
||||
# @@protoc_insertion_point(imports)
|
||||
|
||||
_sym_db = _symbol_database.Default()
|
||||
|
||||
|
||||
from mediapipe.framework import mediapipe_options_pb2 as mediapipe_dot_framework_dot_mediapipe__options__pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/framework/stream_handler.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=_b('\n\032com.google.mediapipe.protoB\022StreamHandlerProto'),
|
||||
serialized_pb=_b('\n(mediapipe/framework/stream_handler.proto\x12\tmediapipe\x1a+mediapipe/framework/mediapipe_options.proto\"\x81\x01\n\x18InputStreamHandlerConfig\x12\x37\n\x14input_stream_handler\x18\x01 \x01(\t:\x19\x44\x65\x66\x61ultInputStreamHandler\x12,\n\x07options\x18\x03 \x01(\x0b\x32\x1b.mediapipe.MediaPipeOptions\"\x9f\x01\n\x19OutputStreamHandlerConfig\x12\x39\n\x15output_stream_handler\x18\x01 \x01(\t:\x1aInOrderOutputStreamHandler\x12\x19\n\x11input_side_packet\x18\x02 \x03(\t\x12,\n\x07options\x18\x03 \x01(\x0b\x32\x1b.mediapipe.MediaPipeOptionsB0\n\x1a\x63om.google.mediapipe.protoB\x12StreamHandlerProto')
|
||||
,
|
||||
dependencies=[mediapipe_dot_framework_dot_mediapipe__options__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
|
||||
_INPUTSTREAMHANDLERCONFIG = _descriptor.Descriptor(
|
||||
name='InputStreamHandlerConfig',
|
||||
full_name='mediapipe.InputStreamHandlerConfig',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='input_stream_handler', full_name='mediapipe.InputStreamHandlerConfig.input_stream_handler', index=0,
|
||||
number=1, type=9, cpp_type=9, label=1,
|
||||
has_default_value=True, default_value=_b("DefaultInputStreamHandler").decode('utf-8'),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='options', full_name='mediapipe.InputStreamHandlerConfig.options', index=1,
|
||||
number=3, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=101,
|
||||
serialized_end=230,
|
||||
)
|
||||
|
||||
|
||||
_OUTPUTSTREAMHANDLERCONFIG = _descriptor.Descriptor(
|
||||
name='OutputStreamHandlerConfig',
|
||||
full_name='mediapipe.OutputStreamHandlerConfig',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='output_stream_handler', full_name='mediapipe.OutputStreamHandlerConfig.output_stream_handler', index=0,
|
||||
number=1, type=9, cpp_type=9, label=1,
|
||||
has_default_value=True, default_value=_b("InOrderOutputStreamHandler").decode('utf-8'),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='input_side_packet', full_name='mediapipe.OutputStreamHandlerConfig.input_side_packet', index=1,
|
||||
number=2, type=9, cpp_type=9, label=3,
|
||||
has_default_value=False, default_value=[],
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='options', full_name='mediapipe.OutputStreamHandlerConfig.options', index=2,
|
||||
number=3, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=233,
|
||||
serialized_end=392,
|
||||
)
|
||||
|
||||
_INPUTSTREAMHANDLERCONFIG.fields_by_name['options'].message_type = mediapipe_dot_framework_dot_mediapipe__options__pb2._MEDIAPIPEOPTIONS
|
||||
_OUTPUTSTREAMHANDLERCONFIG.fields_by_name['options'].message_type = mediapipe_dot_framework_dot_mediapipe__options__pb2._MEDIAPIPEOPTIONS
|
||||
DESCRIPTOR.message_types_by_name['InputStreamHandlerConfig'] = _INPUTSTREAMHANDLERCONFIG
|
||||
DESCRIPTOR.message_types_by_name['OutputStreamHandlerConfig'] = _OUTPUTSTREAMHANDLERCONFIG
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
InputStreamHandlerConfig = _reflection.GeneratedProtocolMessageType('InputStreamHandlerConfig', (_message.Message,), dict(
|
||||
DESCRIPTOR = _INPUTSTREAMHANDLERCONFIG,
|
||||
__module__ = 'mediapipe.framework.stream_handler_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.InputStreamHandlerConfig)
|
||||
))
|
||||
_sym_db.RegisterMessage(InputStreamHandlerConfig)
|
||||
|
||||
OutputStreamHandlerConfig = _reflection.GeneratedProtocolMessageType('OutputStreamHandlerConfig', (_message.Message,), dict(
|
||||
DESCRIPTOR = _OUTPUTSTREAMHANDLERCONFIG,
|
||||
__module__ = 'mediapipe.framework.stream_handler_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.OutputStreamHandlerConfig)
|
||||
))
|
||||
_sym_db.RegisterMessage(OutputStreamHandlerConfig)
|
||||
|
||||
|
||||
DESCRIPTOR._options = None
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@ -0,0 +1,113 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/framework/test_calculators.proto
|
||||
|
||||
import sys
|
||||
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
|
||||
from google.protobuf import descriptor as _descriptor
|
||||
from google.protobuf import message as _message
|
||||
from google.protobuf import reflection as _reflection
|
||||
from google.protobuf import symbol_database as _symbol_database
|
||||
# @@protoc_insertion_point(imports)
|
||||
|
||||
_sym_db = _symbol_database.Default()
|
||||
|
||||
|
||||
from mediapipe.framework import calculator_pb2 as mediapipe_dot_framework_dot_calculator__pb2
|
||||
try:
|
||||
mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe_dot_framework_dot_calculator__options__pb2
|
||||
except AttributeError:
|
||||
mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe.framework.calculator_options_pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/framework/test_calculators.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n*mediapipe/framework/test_calculators.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\xdd\x01\n\x1dRandomMatrixCalculatorOptions\x12\x0c\n\x04rows\x18\x01 \x01(\x05\x12\x0c\n\x04\x63ols\x18\x02 \x01(\x05\x12\x17\n\x0fstart_timestamp\x18\x03 \x01(\x03\x12\x17\n\x0flimit_timestamp\x18\x04 \x01(\x03\x12\x16\n\x0etimestamp_step\x18\x05 \x01(\x03\x32V\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xc8\xa0\xe9\x18 \x01(\x0b\x32(.mediapipe.RandomMatrixCalculatorOptions')
|
||||
,
|
||||
dependencies=[mediapipe_dot_framework_dot_calculator__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
|
||||
_RANDOMMATRIXCALCULATOROPTIONS = _descriptor.Descriptor(
|
||||
name='RandomMatrixCalculatorOptions',
|
||||
full_name='mediapipe.RandomMatrixCalculatorOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='rows', full_name='mediapipe.RandomMatrixCalculatorOptions.rows', index=0,
|
||||
number=1, type=5, cpp_type=1, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='cols', full_name='mediapipe.RandomMatrixCalculatorOptions.cols', index=1,
|
||||
number=2, type=5, cpp_type=1, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='start_timestamp', full_name='mediapipe.RandomMatrixCalculatorOptions.start_timestamp', index=2,
|
||||
number=3, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='limit_timestamp', full_name='mediapipe.RandomMatrixCalculatorOptions.limit_timestamp', index=3,
|
||||
number=4, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='timestamp_step', full_name='mediapipe.RandomMatrixCalculatorOptions.timestamp_step', index=4,
|
||||
number=5, type=3, cpp_type=2, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='ext', full_name='mediapipe.RandomMatrixCalculatorOptions.ext', index=0,
|
||||
number=52056136, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=True, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=96,
|
||||
serialized_end=317,
|
||||
)
|
||||
|
||||
DESCRIPTOR.message_types_by_name['RandomMatrixCalculatorOptions'] = _RANDOMMATRIXCALCULATOROPTIONS
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
RandomMatrixCalculatorOptions = _reflection.GeneratedProtocolMessageType('RandomMatrixCalculatorOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _RANDOMMATRIXCALCULATOROPTIONS,
|
||||
__module__ = 'mediapipe.framework.test_calculators_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.RandomMatrixCalculatorOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(RandomMatrixCalculatorOptions)
|
||||
|
||||
_RANDOMMATRIXCALCULATOROPTIONS.extensions_by_name['ext'].message_type = _RANDOMMATRIXCALCULATOROPTIONS
|
||||
mediapipe_dot_framework_dot_calculator__options__pb2.CalculatorOptions.RegisterExtension(_RANDOMMATRIXCALCULATOROPTIONS.extensions_by_name['ext'])
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@ -0,0 +1,138 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/framework/thread_pool_executor.proto
|
||||
|
||||
import sys
|
||||
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
|
||||
from google.protobuf import descriptor as _descriptor
|
||||
from google.protobuf import message as _message
|
||||
from google.protobuf import reflection as _reflection
|
||||
from google.protobuf import symbol_database as _symbol_database
|
||||
# @@protoc_insertion_point(imports)
|
||||
|
||||
_sym_db = _symbol_database.Default()
|
||||
|
||||
|
||||
from mediapipe.framework import mediapipe_options_pb2 as mediapipe_dot_framework_dot_mediapipe__options__pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/framework/thread_pool_executor.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n.mediapipe/framework/thread_pool_executor.proto\x12\tmediapipe\x1a+mediapipe/framework/mediapipe_options.proto\"\xe9\x02\n\x19ThreadPoolExecutorOptions\x12\x13\n\x0bnum_threads\x18\x01 \x01(\x05\x12\x12\n\nstack_size\x18\x02 \x01(\x05\x12\x1b\n\x13nice_priority_level\x18\x03 \x01(\x05\x12`\n\x1drequire_processor_performance\x18\x04 \x01(\x0e\x32\x39.mediapipe.ThreadPoolExecutorOptions.ProcessorPerformance\x12\x1a\n\x12thread_name_prefix\x18\x05 \x01(\t\"5\n\x14ProcessorPerformance\x12\n\n\x06NORMAL\x10\x00\x12\x07\n\x03LOW\x10\x01\x12\x08\n\x04HIGH\x10\x02\x32Q\n\x03\x65xt\x12\x1b.mediapipe.MediaPipeOptions\x18\x93\xd3\xf5J \x01(\x0b\x32$.mediapipe.ThreadPoolExecutorOptions')
|
||||
,
|
||||
dependencies=[mediapipe_dot_framework_dot_mediapipe__options__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
_THREADPOOLEXECUTOROPTIONS_PROCESSORPERFORMANCE = _descriptor.EnumDescriptor(
|
||||
name='ProcessorPerformance',
|
||||
full_name='mediapipe.ThreadPoolExecutorOptions.ProcessorPerformance',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='NORMAL', index=0, number=0,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='LOW', index=1, number=1,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='HIGH', index=2, number=2,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=332,
|
||||
serialized_end=385,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_THREADPOOLEXECUTOROPTIONS_PROCESSORPERFORMANCE)
|
||||
|
||||
|
||||
_THREADPOOLEXECUTOROPTIONS = _descriptor.Descriptor(
|
||||
name='ThreadPoolExecutorOptions',
|
||||
full_name='mediapipe.ThreadPoolExecutorOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='num_threads', full_name='mediapipe.ThreadPoolExecutorOptions.num_threads', index=0,
|
||||
number=1, type=5, cpp_type=1, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='stack_size', full_name='mediapipe.ThreadPoolExecutorOptions.stack_size', index=1,
|
||||
number=2, type=5, cpp_type=1, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='nice_priority_level', full_name='mediapipe.ThreadPoolExecutorOptions.nice_priority_level', index=2,
|
||||
number=3, type=5, cpp_type=1, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='require_processor_performance', full_name='mediapipe.ThreadPoolExecutorOptions.require_processor_performance', index=3,
|
||||
number=4, type=14, cpp_type=8, label=1,
|
||||
has_default_value=False, default_value=0,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='thread_name_prefix', full_name='mediapipe.ThreadPoolExecutorOptions.thread_name_prefix', index=4,
|
||||
number=5, type=9, cpp_type=9, label=1,
|
||||
has_default_value=False, default_value=_b("").decode('utf-8'),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='ext', full_name='mediapipe.ThreadPoolExecutorOptions.ext', index=0,
|
||||
number=157116819, type=11, cpp_type=10, label=1,
|
||||
has_default_value=False, default_value=None,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=True, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
_THREADPOOLEXECUTOROPTIONS_PROCESSORPERFORMANCE,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=107,
|
||||
serialized_end=468,
|
||||
)
|
||||
|
||||
_THREADPOOLEXECUTOROPTIONS.fields_by_name['require_processor_performance'].enum_type = _THREADPOOLEXECUTOROPTIONS_PROCESSORPERFORMANCE
|
||||
_THREADPOOLEXECUTOROPTIONS_PROCESSORPERFORMANCE.containing_type = _THREADPOOLEXECUTOROPTIONS
|
||||
DESCRIPTOR.message_types_by_name['ThreadPoolExecutorOptions'] = _THREADPOOLEXECUTOROPTIONS
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
ThreadPoolExecutorOptions = _reflection.GeneratedProtocolMessageType('ThreadPoolExecutorOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _THREADPOOLEXECUTOROPTIONS,
|
||||
__module__ = 'mediapipe.framework.thread_pool_executor_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.ThreadPoolExecutorOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(ThreadPoolExecutorOptions)
|
||||
|
||||
_THREADPOOLEXECUTOROPTIONS.extensions_by_name['ext'].message_type = _THREADPOOLEXECUTOROPTIONS
|
||||
mediapipe_dot_framework_dot_mediapipe__options__pb2.MediaPipeOptions.RegisterExtension(_THREADPOOLEXECUTOROPTIONS.extensions_by_name['ext'])
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@ -0,0 +1,115 @@
|
||||
import csv
|
||||
import random
|
||||
import tensorflow as tf
|
||||
import os
|
||||
column_names = []
|
||||
|
||||
train_feature=[]
|
||||
train_label=[]
|
||||
test_feature=[]
|
||||
test_label=[]
|
||||
|
||||
with open("test.csv","r") as r:
|
||||
reader = csv.reader(r)
|
||||
for i in reader:
|
||||
rand = random.randint(0,100)
|
||||
if rand<=85:
|
||||
train_label.append((eval(i.pop())))
|
||||
b = [eval(j) for j in i]
|
||||
train_feature.append(b)
|
||||
else:
|
||||
test_label.append((eval(i.pop())))
|
||||
b = [eval(j) for j in i]
|
||||
test_feature.append(b)
|
||||
train_features = tf.constant(train_feature)
|
||||
train_labels = tf.constant(train_label)
|
||||
test_features = tf.constant(test_feature)
|
||||
test_labels = tf.constant(test_label)
|
||||
|
||||
|
||||
model = tf.keras.Sequential([
|
||||
tf.keras.layers.Flatten(input_shape=(252,)),
|
||||
tf.keras.layers.Dense(10, activation=tf.nn.relu), # input shape required
|
||||
tf.keras.layers.Dense(10, activation=tf.nn.relu),
|
||||
tf.keras.layers.Dense(6)
|
||||
])
|
||||
predictions = model(train_features)
|
||||
tf.nn.softmax(predictions[:5])
|
||||
print("Prediction: {}".format(tf.argmax(predictions, axis=1)))
|
||||
print(" Labels: {}".format(train_labels))
|
||||
loss_object = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
|
||||
|
||||
def loss(model, x, y, training):
|
||||
# training=training is needed only if there are layers with different
|
||||
# behavior during training versus inference (e.g. Dropout).
|
||||
y_ = model(x, training=training)
|
||||
|
||||
return loss_object(y_true=y, y_pred=y_)
|
||||
l = loss(model, train_features, train_labels, training=False)
|
||||
def grad(model, inputs, targets):
|
||||
with tf.GradientTape() as tape:
|
||||
loss_value = loss(model, inputs, targets, training=True)
|
||||
return loss_value, tape.gradient(loss_value, model.trainable_variables)
|
||||
|
||||
optimizer = tf.keras.optimizers.SGD(learning_rate=0.01)
|
||||
loss_value, grads = grad(model, train_features, train_labels)
|
||||
|
||||
print("Step: {}, Initial Loss: {}".format(optimizer.iterations.numpy(),
|
||||
loss_value.numpy()))
|
||||
|
||||
optimizer.apply_gradients(zip(grads, model.trainable_variables))
|
||||
|
||||
print("Step: {}, Loss: {}".format(optimizer.iterations.numpy(),
|
||||
loss(model, train_features, train_labels, training=True).numpy()))
|
||||
|
||||
## Note: Rerunning this cell uses the same model variables
|
||||
|
||||
# Keep results for plotting
|
||||
train_loss_results = []
|
||||
train_accuracy_results = []
|
||||
|
||||
num_epochs = 351
|
||||
|
||||
for epoch in range(num_epochs):
|
||||
epoch_loss_avg = tf.keras.metrics.Mean()
|
||||
epoch_accuracy = tf.keras.metrics.SparseCategoricalAccuracy()
|
||||
|
||||
# Training loop - using batches of 32
|
||||
branch = 32
|
||||
for num in range(len(train_features)//branch):
|
||||
x = train_features[num*branch:(num+1)*branch]
|
||||
y = train_labels[num*branch:(num+1)*branch]
|
||||
# Optimize the model
|
||||
loss_value, grads = grad(model, x, y)
|
||||
optimizer.apply_gradients(zip(grads, model.trainable_variables))
|
||||
|
||||
# Track progress
|
||||
epoch_loss_avg.update_state(loss_value) # Add current batch loss
|
||||
# Compare predicted label to actual label
|
||||
# training=True is needed only if there are layers with different
|
||||
# behavior during training versus inference (e.g. Dropout).
|
||||
epoch_accuracy.update_state(y, model(x, training=True))
|
||||
|
||||
# End epoch
|
||||
train_loss_results.append(epoch_loss_avg.result())
|
||||
train_accuracy_results.append(epoch_accuracy.result())
|
||||
|
||||
if epoch % 50 == 0:
|
||||
print("Epoch {:03d}: Loss: {:.3f}, Accuracy: {:.3%}".format(epoch,
|
||||
epoch_loss_avg.result(),
|
||||
epoch_accuracy.result()))
|
||||
|
||||
test_accuracy = tf.keras.metrics.Accuracy()
|
||||
model.save("model")
|
||||
|
||||
for num in range(len(test_features)):
|
||||
# training=False is needed only if there are layers with different
|
||||
# behavior during training versus inference (e.g. Dropout).
|
||||
x = test_features[num:num+1]
|
||||
y = test_labels[num:num+1]
|
||||
logits = model(x, training=False)
|
||||
prediction = tf.argmax(logits, axis=1, output_type=tf.int32)
|
||||
print("真实值为",y,"预测值为",prediction)
|
||||
test_accuracy(prediction, y)
|
||||
|
||||
print("Test set accuracy: {:.3%}".format(test_accuracy.result()))
|
||||
Loading…
Reference in new issue