Skip to content

***Test of mask_rcnn_inception_v2_coco*** #4689

@SpiralBeing

Description

@SpiralBeing

System information

What is the top-level directory of the model you are using: /home/jesse/gpu-py3/models/research/object_detection
Have I written custom code (NO):
OS Platform and Distribution (Windows 10):
TensorFlow installed from (binary[using pip3):
TensorFlow version (1.8.0):
Python version: (3.6.5)
Bazel version (N/A):
GCC/Compiler version (N/A):
CUDA/cuDNN version: CUDA 9.0/ CUDDN 7
GPU model and memory: Nvidia GTX 1050/ 2GB
Exact command to reproduce:
python train.py --logtostderr --train_dir=data/ --pipeline_config_path=data/mask_rcnn_inception_v2_coco.config

When I'm creating tf__pet_record I see mask as and in tutorial (feature_dict["image/object/mask"] bytes_list {
value: "\211PNG\r\n\032\n\000\000\000\rIHDR......)
(I change also here
if not faces_only: # Original is ''if not faces_only''
mask_remapped = (mask_np == 1).astype(np.uint8) ) #==1 (in original !=2)
Input image for training mask were png image (png with pixels ==1 or 0, with 3 channels)

I've trained model (during the training process I see that box and mask were training [double step]) but in jupyter notebook I can only observe box in the image without mask..
In jyputer:
tensor_dict {'num_detections': <tf.Tensor 'num_detections:0' shape= dtype=float32>, 'detection_boxes': <tf.Tensor 'detection_boxes:0' shape= dtype=float32>, 'detection_scores': <tf.Tensor 'detection_scores:0' shape= dtype=float32>, 'detection_classes': <tf.Tensor 'detection_classes:0' shape= dtype=float32>} Mask isn't here!
Must I exact something other additional command except this (python train.py --logtostderr --train_dir=data/ --pipeline_config_path=data/mask_rcnn_inception_v2_coco.config) ?

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions