-
Notifications
You must be signed in to change notification settings - Fork 45.5k
Description
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) ?