TexasInstruments / edgeai-yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite. Forked from https://ultralytics.com/yolov5
https://github.com/TexasInstruments/edgeai
GNU General Public License v3.0
655 stars 120 forks source link

pretrained model is wrong #98

Open zcunyi opened 1 year ago

zcunyi commented 1 year ago

❔Question

yolov5s6-pose pretrained model: http://software-dl.ti.com/jacinto7/esd/modelzoo/gplv3/08_02_00_11/edgeai-yolov5/pretrained_models/checkpoints/keypoint/coco/edgeai-yolov5/other/person_detector/yolov5s6_960_71p6_93p1/weights/last.pt

but the weights in opt.yaml is /user/a0132471/Files/bit-bucket/pytorch/yolov5_pose/runs/train/shared_models/person_detector/yolov5s6_person_64p8_90p2/weights/last.pt

I used the pretrained model to obtain map0.547, which is lower than the readme 0.607

Additional context

zcunyi commented 1 year ago

d