Hi Jaka jakaskerl
Yes I believe so.
This is my script set-up:
pipeline = dai.Pipeline()
#### det NN ###
nn_det = pipeline.create(dai.node.YoloDetectionNetwork)
nn_det.setConfidenceThreshold(0.4)
nn_det.setBlobPath('/Users/cloudscapespare/Documents/yolov5s_seg/224_x/best_openvino_2022.1_8shave.blob')
nn_det.input.setBlocking(True)
nn_det.setNumClasses(6)
nn_det.setCoordinateSize(4)
nn_det.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])
nn_det.setAnchorMasks({"side28": [0, 1, 2], "side14": [3, 4, 5], "side7": [6, 7, 8]})This is my script set up:
While this is what's in my JSON file:
"nn_config": {
"output_format": "detection",
"NN_family": "YOLO",
"input_size": "224x224",
"NN_specific_metadata": {
"classes": 6,
"coordinates": 4,
"anchors": [
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
],
"anchor_masks": {
"side28": [
0,
1,
2
],
"side14": [
3,
4,
5
],
"side7": [
6,
7,
8
]
},
"iou_threshold": 0.5,
"confidence_threshold": 0.5
}
Don't think I need to change anything in my script, right?
Or it might be a issue about optimisation?
Regards,
Austin