Did you know how to train the model on the custom datasets?

[kangkang59812]

I want to use the model to segment food. And I just need a picture to segment.

[toniz]

Use this model From model zoo:

mobilenetv2_dm05_coco_voc_trainaug
https://github.com/tensorflow/models/blob/master/research/deeplab/g3doc/model_zoo.md

Convert tflite like this:
convert2pb.sh:

xwj@master01:/data/xwj/segmentation/train$ cat convert2pb.sh
python ../../models/research/deeplab/export_model.py
--checkpoint_path=log_all/model.ckpt-100000
--depth_multiplier=0.5
--quantize_delay_step=0
--export_path=tflite_all/frozen_inference_graph.pb

conver2tflite.sh:

xwj@master01:/data/xwj/segmentation/train$ cat conver2tflite.sh
tflite_convert
--graph_def_file=tflite_all/frozen_inference_graph.pb
--output_file=tflite_all/frozen_inference_graph.tflite
--output_format=TFLITE
--input_shape=1,513,513,3
--input_arrays="MobilenetV2/MobilenetV2/input"
--change_concat_input_ranges=true
--output_arrays="ArgMax"

[hemant-novotrax]

How to colored or remove background?