使用预训练权重呢,一般是为了缩短网络的训练时间,达到更好的精度。 我们去GitHubReleases · ultralytics/yolov5 (github.com)找到对应版本的预训练权重 yolov5s.pt,放到weights目录下。 数据文件配置 在data下找到voc.yaml,复制一份再命名为mask.yaml,将其修改为: # download command/URL (optional) #download: ...
在examples目录下,复制ax_yolov5s_steps.cc另存为ax_yolov5s_my.cc 修改ax-samples/examples/base/detection.hpp,新增一个generate_proposals_n函数,cls_num就是模型类别数 staticvoidgenerate_proposals_n(intcls_num,intstride, constfloat* feat,floatprob_threshold, std::vector<Object>& objects,intletterbox...