update input vars
This commit is contained in:
parent
211c5dbf8e
commit
c2b81d4b6c
|
@ -1,36 +0,0 @@
|
|||
# Contributing guidelines
|
||||
|
||||
## What type of pull request do we accept into MediaPipe repository?
|
||||
|
||||
* Bug fixes
|
||||
* Documentation fixes
|
||||
|
||||
For new feature additions (e.g., new graphs and calculators), we are currently not planning to accept new feature pull requests into the MediaPipe repository. Instead, we like to get contributors to create their own repositories of the new feature and list it at [Awesome MediaPipe](https://mediapipe.org). This will allow contributors to more quickly get their code out to the community.
|
||||
|
||||
Before sending your pull requests, make sure you followed this list.
|
||||
|
||||
- Read [contributing guidelines](CONTRIBUTING.md).
|
||||
- Read [Code of Conduct](CODE_OF_CONDUCT.md).
|
||||
- Ensure you have signed the [Contributor License Agreement (CLA)](https://cla.developers.google.com/).
|
||||
|
||||
## How to become a contributor and submit your own code
|
||||
|
||||
### Contributor License Agreements
|
||||
|
||||
We'd love to accept your patches! Before we can take them, we have to jump a couple of legal hurdles.
|
||||
|
||||
Please fill out either the individual or corporate Contributor License Agreement (CLA).
|
||||
|
||||
* If you are an individual writing original source code and you're sure you own the intellectual property, then you'll need to sign an [individual CLA](https://code.google.com/legal/individual-cla-v1.0.html).
|
||||
* If you work for a company that wants to allow you to contribute your work, then you'll need to sign a [corporate CLA](https://code.google.com/legal/corporate-cla-v1.0.html).
|
||||
|
||||
Follow either of the two links above to access the appropriate CLA and instructions for how to sign and return it. Once we receive it, we'll be able to accept your pull requests.
|
||||
|
||||
***NOTE***: Only original source code from you and other people that have signed the CLA can be accepted into the main repository.
|
||||
|
||||
### Contributing code
|
||||
|
||||
If you have bug fixes and documentation fixes to MediaPipe, send us your pull requests! For those
|
||||
just getting started, GitHub has a [howto](https://help.github.com/articles/using-pull-requests/).
|
||||
|
||||
MediaPipe team members will be assigned to review your pull requests. Once the bug/documentation fixes are verified, a MediaPipe team member will acknowledge your contribution in the pull request comments, manually merge the fixes into our internal codebase upstream, and apply the `to be closed` label to the pull request. These fixes will later be pushed to GitHub in the next release, and a MediaPipe team member will then close the pull request.
|
|
@ -143,10 +143,10 @@ Note: This currently works only on Linux, and please first follow
|
|||
|
||||
## Running the pretrained model for sign language recognition
|
||||
|
||||
Insert your data to any folder and pass it as a ```--output_data_path``` variable. The result will appear in the same folder and will be called `result.txt`.
|
||||
Insert your data to any folder and pass it as a ```--processed_data_path``` variable. The result will appear in the same folder and will be stored in the `result.txt`.
|
||||
|
||||
```bash
|
||||
cd RNN/
|
||||
cd sign_prediction/
|
||||
|
||||
python3 predict.py --input_data_path='./test_video/' --output_data_path='./test_video_output/'
|
||||
python3 predict.py --processed_data_path='./test_video_output/'
|
||||
```
|
||||
|
|
|
@ -1,87 +0,0 @@
|
|||
import os
|
||||
import numpy as np
|
||||
import tensorflow as tf
|
||||
import argparse
|
||||
|
||||
def load_data(dirname):
|
||||
if dirname[-1] != '/':
|
||||
dirname=dirname+'/'
|
||||
listfile=os.listdir(dirname)
|
||||
X = []
|
||||
Y = []
|
||||
for text in listfile:
|
||||
textname = dirname + text
|
||||
numbers=[]
|
||||
with open(textname, mode = 'r') as t:
|
||||
numbers = [float(num) for num in t.read().split()]
|
||||
for i in range(len(numbers),25200):
|
||||
numbers.extend([0.000])
|
||||
landmark_frame=[]
|
||||
row=0
|
||||
for i in range(0,70):
|
||||
landmark_frame.extend(numbers[row:row+84])
|
||||
row += 84
|
||||
landmark_frame=np.array(landmark_frame)
|
||||
landmark_frame=landmark_frame.reshape(-1,84)
|
||||
X.append(np.array(landmark_frame))
|
||||
Y.append(text)
|
||||
X = np.array(X)
|
||||
Y = np.array(Y)
|
||||
# print(Y)
|
||||
x_train = X
|
||||
x_train=np.array(x_train)
|
||||
return x_train,Y
|
||||
|
||||
def load_label():
|
||||
listfile=[]
|
||||
with open("label.txt",mode='r') as l:
|
||||
listfile=[i for i in l.read().split()]
|
||||
label = {}
|
||||
count = 1
|
||||
for l in listfile:
|
||||
if "_" in l:
|
||||
continue
|
||||
label[l] = count
|
||||
count += 1
|
||||
return label
|
||||
|
||||
def main(input_data_path, output_data_path):
|
||||
comp = 'bazel build -c opt --define MEDIAPIPE_DISABLE_GPU=1 \
|
||||
mediapipe/examples/desktop/hand_tracking:hand_tracking_cpu'
|
||||
|
||||
cmd = 'GLOG_logtostderr=1 bazel-bin/mediapipe/examples/desktop/hand_tracking/hand_tracking_cpu \
|
||||
--calculator_graph_config_file=mediapipe/graphs/hand_tracking/hand_tracking_desktop_live.pbtxt'
|
||||
|
||||
output_dir = output_data_path
|
||||
x_test, Y = load_data(output_dir)
|
||||
new_model = tf.keras.models.load_model('model.h5')
|
||||
# new_model.summary()
|
||||
|
||||
labels=load_label()
|
||||
xhat = x_test
|
||||
print(xhat)
|
||||
|
||||
yhat = new_model.predict(xhat)
|
||||
|
||||
print(yhat)
|
||||
predictions = np.array([np.argmax(pred) for pred in yhat])
|
||||
rev_labels = dict(zip(list(labels.values()), list(labels.keys())))
|
||||
s=0
|
||||
filel = np.array(Y)
|
||||
txtpath=output_data_path+"result.txt"
|
||||
with open(txtpath, "w") as f:
|
||||
for i in predictions:
|
||||
f.write(Y[s])
|
||||
f.write(" ")
|
||||
f.write(rev_labels[i])
|
||||
f.write("\n")
|
||||
s+=1
|
||||
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(description='Predict Sign language with Mediapipe')
|
||||
parser.add_argument("--input_data_path",help=" ")
|
||||
parser.add_argument("--output_data_path",help=" ")
|
||||
args = parser.parse_args()
|
||||
input_data_path = args.input_data_path
|
||||
output_data_path = args.output_data_path
|
||||
main(input_data_path,output_data_path)
|
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user