Hey! So for some time now, i've been working on using the Oak-D Lite camera, and i've successfully managed to incorporate a YoloV5 model. The issue is that the original model that i've made is of size 416x416 and the image returned is a bit small. For some time now i've been trying to re-train the model using the notebook you provided, with slightly larger image size but for some reason it simply does not work. The pipeline starts regularly, and the preview image is indeed larger but the following error keeps occuring in the console: [1844301041DDC21200] [344.188] [SpatialDetectionNetwork(1)] [error] Mask is not defined for output layer with width '80'. Define at pipeline build time using: 'setAnchorMasks' for 'side80'. [1844301041DDC21200] [344.188] [SpatialDetectionNetwork(1)] [error] Mask is not defined for output layer with width '40'. Define at pipeline build time using: 'setAnchorMasks' for 'side40'. [1844301041DDC21200] [344.188] [SpatialDetectionNetwork(1)] [error] Mask is not defined for output layer with width '20'. Define at pipeline build time using: 'setAnchorMasks' for 'side20'. The anchors and anchor masks in the cfg json are as following : ` "anchors": [ 10, 13, 16, 30, 33, 23, 30, 61, 62, 45, 59, 119, 116, 90, 156, 198, 373, 326 ], "anchor_masks": { "side52": [0, 1, 2], "side26": [3, 4, 5], "side13": [6, 7, 8] },` What could be the issue here? Any help is appriciated!

Hello @"medonni"#p3883 , I believe you need to copy anchors/masks from the generated json (or just use that json) when following the [yolov5 training notebook](https://github.com/luxonis/depthai-ml-training/blob/master/colab-notebooks/YoloV5_training.ipynb). Have you taken that step? Thanks, Erik

@"erik"#p3886 Hey Erik! Which json are you exactly referring to? The only mentions of json in the notebook are `save_json=False` in the validation step, and in the on-device decoding where its explained that we should only just change num of classes and labels. What am i missing here? Thanks in advance!

Hello @"medonni"#p3894 , I have reread your initial post and I am not sure, have you changed the default resolution (416x416)? If you haven't, the JSON at the depthai-experiments should work as expected. Thanks, Erik

@"erik"#p3895 Hey! Yea, i'm trying to change the model to a bigger size, currently trying a 640x640. The first model i trained was at 416x416 and it worked fine with the provided JSON. The issue is that from what i realized, the model size is directly tied to the size of preview window and it being at 416x416 is just a tad bit small, therefore i want to up the size a little bit.

Resolved thanks to Conor on discord. Issue was that anchor masks change based on the actual model size, so for 640x640 i needed 80/40/20 masks in the config. To calculate it we should do size/32, size/16, size/8, and use the obtained anchor masks.

@"medonni"#p3901 Please can you share what your final JSON file looked like? I can't figure this out...

@"stephansturges"#p5624 Hey! sorry for the late reply. Here is the JSON as of today. `{ "nn_config": { "output_format": "detection", "NN_family": "YOLO", "input_size" : "416x416", "NN_specific_metadata": { "classes": 1, "coordinates": 4, "anchors": [ 10, 13, 16, 30, 33, 23, 30, 61, 62, 45, 59, 119, 116, 90, 156, 198, 373, 326 ], "anchor_masks" : { "side52": [0, 1, 2], "side26": [3, 4, 5], "side13": [6, 7, 8] }, "iou_threshold": 0.5, "confidence_threshold": 0.65 } }, "mappings": { "labels": ["plate"] } } ` Hope this helps!

Hello @"medonni"#p5686 @"stephansturges"#p5699 I have tried suggested solution it's not working for me. Can you send me working JSON file for custom yolov5 model ... **ERROR** - [DetectionNetwork(1)] [error] Mask is not defined for output layer with width '6'. Define at pipeline build time using: 'setAnchorMasks' for 'side6'

@"StavanRupareliya"#p6297 Unfortunately not, as i do not have the access to it anymore. What is the model size you are trying to do?

Changing model size when training returns output layer width error

erik

krishnashravan you need to define masks/anchors, you currently have this empty:

"anchors": [],
"anchor_masks": {},

If you have .pt model, you can use tools.luxonis.com, which will return the json together with the anchors/masks.

krishnashravan

I used the tools to get the blob file along with the json file thats the json file from the luxonis tool

erik

krishnashravan could you provide .pt?

ThasnimolVSam22d007

erik in luxonics there is no option to provide .pt in https://blobconverter.luxonis.com/ - in this tool

krishnashravan

the pt files
the hand gesture that is nano and small model trained with 416 input size (handntrained and handstrained)
the person detection that is nano trained with 640 input size (personntrained)

erik

krishnashravan Have you also posted the question on their webpage?

krishnashravan

erik

No not yet

Just now i tried with the yolov8 training colab file from the repo i'm getting the same empty anchor values

krishnashravan

erik
these are the pt files trained with the YoloV8_training.ipynb notebook

Timmm

I have also trained YOLOv8, I think it YOLOv8 doesnot use anchor boxes instead uses Dynamic Convolutional layers. krishnashravan. Make sure you double check to use handntrained and handstrained model with 416x416 input image size.

\';LKMJNH BVCXP

Timmm

detectionNetwork.setAnchors( [10.0, 13.0, 16.0, 30.0, 33.0, 23.0, 30.0, 61.0, 62.0, 45.0, 59.0, 119.0, 116.0, 90.0, 156.0, 198.0, 373.0,326.0])

detectionNetwork.setAnchorMasks({"side52": [0, 1, 2], "side26": [3, 4, 5], "side13": [6, 7, 8]})

@krishnashravan Try setting your anchors and anchormasks as above.

ThasnimolVSam22d007

erik

how to change this default resolutuon?

and get this json file?

erik

Hi ThasnimolVSam22d007 ,
Size of the model has to be set/configured before using blobconverter.

« Previous Page