Skip to content
This repository has been archived by the owner on Sep 16, 2024. It is now read-only.

deeplab predict a blank segmentation masks #23

Open
hakS07 opened this issue Jun 10, 2019 · 0 comments
Open

deeplab predict a blank segmentation masks #23

hakS07 opened this issue Jun 10, 2019 · 0 comments

Comments

@hakS07
Copy link

hakS07 commented Jun 10, 2019

@DrSleep Hi sir, I wish you can help me with this,

System information

What is the top-level directory of the model you are using:
https://github.com/tensorflow/models/tree/master/research/deeplab
** Have I written custom code (as opposed to using a stock example script provided in TensorFlow):**
No
**OS Platform and Distribution (e.g., Linux Ubuntu 16.04):**ubuntu 18.04
**TensorFlow installed from (source or binary):**binary
**TensorFlow version (use command below):**1.13.1 CPU
**Bazel version (if compiling from source):**N/A
**CUDA/cuDNN version:**9.2/7.1
**GPU model and memory:**GeForce GTX 970M

Exact command to reproduce:
python train.py \ --logtostderr \ --vis_split = "train" \ --model_variant = "xception_65" \ --atrous_rates = 6 \ --atrous rates = 12 \ --atrous rates = 18 \ --output_stride = 16 \ --decoder_output_stride = 4 \ --training_number_of_steps = 1000 --train_crop_size = 513 \ --train_batch_size = 1 \ --train_crop_size = 513 \ --Fine_tune_batch_norm=False \ --Tf_initial_checkpoint = "./ Data / Init_models / Deelabv3_pascal_train_aug \ model.ckpt" --Initialize_last_layer = False \ --Last_layers_contain_logits_only = True \ --train_logdir="./data/log/train" \ --dataset_dir="./data/tfrecord" \ --dataset="pascal_voc_seg"

Describe the problem

Hi i trained deeplab model and does not predict anything just a black background, i don't know what's the problem
-training on custom dataset (iris data)
-data = RGB image + ( 0-1) label: 400 * 300
-classe=2
-convert to record format:
-training step:
==> Loss = 0.2 ~ 0.1
-convert to .pb step
python export_model.py \ --logtostderr \ -model_variant = "xception_65" \ --atrous_rates = 6 \ --atrous_rates = 12 \ --atrous_rates = 18 \ --output_stride = 16
until this step everything looks fine
-test step
Capture
1
2
3

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant