Train tf lite object-detection model - #68

Open

kickermeister wants to merge 47 commits into

get_training_data

Commits on Aug 28, 2021

extended format conversion function and filter only real ingredients
kickermeister
committed
a998c6f
fix: tf make_model expects x_max / y_max and not width / heigth of bounding box
kickermeister
committed
efa72e2

Commits on Aug 29, 2021

added a few functions to create dataset conveniently
kickermeister
committed
2df347d

Commits on Aug 30, 2021

some changes and imports for trainer.py
kickermeister
committed
69cb492
Merge branch 'tsflite_model' into get_training_data
kickermeister
committed
2dae8d9
added some initial basics to the trainer
kickermeister
committed
29bf51c
make tensorflow less verbose
kickermeister
committed
b65f61d
add missing dependency and make tf less verbose
kickermeister
committed
a42f4af
smaller classes list added
kickermeister
committed
08d116f
ingredient classes can be passed to functions
kickermeister
committed
c2cea81
fix: variable name
kickermeister
committed
cc57c7c
fix: use correct parameter name
kickermeister
committed
3d59300
fix parameter name
kickermeister
committed
cf9b948
fixed baseurl for images and model type
kickermeister
committed
2aaf8f9
fix path
kickermeister
committed
ace1592
add gcloud training command to Makefile
kickermeister
committed
f4fdd39
specify tensorflow version for gcloud AI platform training
kickermeister
committed
f783000
fix tf version
kickermeister
committed
145dcd2
function added to create randomly selected slice of datasets with balanced food classes
kickermeister
committed
55dde85
fixed a few things and added __main__ part for gcloud training
kickermeister
committed
451ea86

Commits on Aug 31, 2021

changed a few names
kickermeister
committed
13ead4a
fix imports
kickermeister
committed
14b753e
Merge branch 'master' into get_training_data
kickermeister
committed
9f3fd15
use trained tflite model for metrics and API
kickermeister
committed
99e9cb7
enable a few more parameters for trainer
kickermeister
committed
e1c4d00
added the class labels map dictionary
kickermeister
committed
562d2c7
add function to load from datasets from cache
kickermeister
committed
08bbe16
fixed load from cache and enable automatic loading form local folder when folder exists
kickermeister
committed
edfc6af
fix condition
kickermeister
committed
ecd6df0
fix condition
kickermeister
committed
091580d
save class labels separately as pickle file
kickermeister
committed
b30de1d
enabled trainer parameters as cli arguments
kickermeister
committed
dbbc9e3
pass cli arguments to run method
kickermeister
committed
b5cd3ef
changed default model spec to lite0 again to avoid potential / likely issues
kickermeister
committed
1dd0a38
fix: add missing default value for spec
kickermeister
committed
108f853
fix parameter types
kickermeister
committed
21e0476
fix bool type action
kickermeister
committed
a00954f
revert bool action type
kickermeister
committed
76a353a
fix: filename of class labels
kickermeister
committed
b4432cf
fix export pickel name
kickermeister
committed
12e7664

Commits on Sep 2, 2021