1、我想要替换成自己的权重文件,在launch文件中修改yolo.pt为best.pt
2、在~/yolo_fusion/src/src/detect.py文件中修改yolo.pt的路径为best.pt,具体代码在51,52,219,220行
3、修改完成之后source之后会出现报错,这里采用博主的解决办法,具体的路径为
/home/fyy/anaconda3/lib/python3.10/site-packages/torch/nn/modules/upsampling.py
修改的代码行为155到160行
4、我改了~/yolo_fusion/yolov5/data/coco.yaml的第20行为tinycar,原来为car。