• Hardware A5000 • Network Type Mask_rcnn • TLT Version nvidia/tao/tao-toolkit-tf: v3.22.05-tf1.15.5-py3 • Training spec file tao_maskrcnn_02_09_24_train_v6.t…

Training with less dataset is in order to narrow down the issue. One potential issue in your spec file is that you only set num_classes: 3 . But you mentioned that your dataset is a derivative of the MS coco dataset filtered to include only 3 classes. In this case, please set num_classes: 4. See h…

Low accuracy for MS COCO dataset in tao maskrcnn model training

Accelerated Computing Intelligent Video Analytics TAO Toolkit

Morganh September 11, 2024, 7:29am 2

Please refer to the spec in Poor metric results after retraining maskrcnn using TLT notebook - #17 by Morganh.

Topic		Replies	Views
Tao mask_rcnn training exits with NaN loss TAO Toolkit	12	101	September 18, 2024
Train mask-rcnn failure TAO Toolkit tao	16	1201	November 25, 2021
Mask R-CNN stops abruptly while training using custom coco dataset TAO Toolkit	12	1411	February 22, 2022
Training problems with the TAO toolkit using Mask R-CNN TAO Toolkit	7	878	November 22, 2022
MaskRCNN Input to reshape is a tensor with 3135248 values, but the requested shape has 2691200 TAO Toolkit	38	1211	May 9, 2023
Model diverged with loss = nan TAO Toolkit	2	594	May 16, 2023
Nvidia TAO Maskrcnn training problem. Training is complete but Evaluation Metrics are all 0, and can not achieve right mask on inferences TAO Toolkit	6	530	May 5, 2023
TensorflowGPU- MaskRCNN loss is Nan values only Computer Vision & Image Processing	1	1083	February 12, 2023
Mask_rcnn shows training logs loss 0.00000 fast_rcnn class loss: 0.00000 fast_rcnn box loss: 0.00000 TAO Toolkit	22	824	August 16, 2022
FileNotFoundError: [Errno 2] No such file or directory: '/home/ubuntu/getting_started_v5.0.0/notebooks/tao_launcher_starter_kit/mask_rcnn/specs/maskrc TAO Toolkit python	44	1651	September 5, 2023

Low accuracy for MS COCO dataset in tao maskrcnn model training

Related topics