Quantcast
Channel: TAO Toolkit - NVIDIA Developer Forums
Viewing all articles
Browse latest Browse all 541

Black Mask Images While Converting COCO to UNET

$
0
0

Hey,
I want to train UNET-18 using tao on my custom dataset. I have my dataset in COCO format with jpg images and i went through this post where i found the following command to convert the COCO Dataset Format to UNET

!tao model unet dataset_convert -f "/workspace/tao-experiments/data/new_merged_annotations.json" -r "/workspace/tao-experiments/results"

The issue is when I execute this command it generates the mask images with black screen only and in logs it shows the status as PASS.

The mask image is just a black image without any mask in it.
Can you help me how can i resolve this issue and create the exact masks to train the UNET.

1 post - 1 participant

Read full topic


Viewing all articles
Browse latest Browse all 541

Trending Articles