Yolov5s/ai_training/regression/litehrnet/utils/yolov5_init_params.yaml

16 lines
710 B
YAML

model_path: /mnt/models/Object_models/YOLOv5/101221_yolov5/yolov5s_noupsample_LUMENS_v5_480x256.onnx
model_id: 260
yaml_path: /mnt/models/Object_models/YOLOv5/101221_yolov5/yolov5s-noupsample-2class.yaml
grid20_path: /mnt/models/Object_models/YOLOv5/101221_yolov5/20_480x256.npy
grid40_path: /mnt/models/Object_models/YOLOv5/101221_yolov5/40_480x256.npy
grid80_path: /mnt/models/Object_models/YOLOv5/101221_yolov5/80_480x256.npy
num_classes: 2
input_shape: [256, 480]
conf_thres: 0.3
iou_thres: 0.5
top_k_num: 3000
detection_type: Kneron_detection_field
vanish_point: 0.0
label_mapping: [15,56]
class_name: ["person","head"]
anchors: [[5,6, 11,13, 18,22],[19,51, 31,39, 37,95], [60,161, 95,256, 231,458]]