YOLOv7-obb复现过程记录

大佬的代码git下载代码

下载ssdd数据集：

SSDD数据集下载地址如下
链接: 百度网盘 请输入提取码 提取码: 2021（原博客）

根据readme解决报错voc_annotation.py所报错误转换好数据格式后运行train.py训练到10/100epoch 报错，报错提示到如下：

Get map.
Traceback (most recent call last):
File "d:/yolov7-obb/train.py", line 552, in <module>
fit_one_epoch(model_train, model, ema, yolo_loss, loss_history, eval_callback, optimizer, epoch, epoch_step, epoch_step_val, gen, gen_val, UnFreeze_Epoch, Cuda, fp16, scaler, save_period, save_dir, local_rank)
File "d:\yolov7-obb\utils\utils_fit.py", line 104, in fit_one_epoch
eval_callback.on_epoch_end(epoch + 1, model_train_eval)
File "d:\yolov7-obb\utils\callbacks.py", line 206, in on_epoch_end
self.get_map_txt(image_id, image, self.class_names, self.map_out_path)
File "d:\yolov7-obb\utils\callbacks.py", line 142, in get_map_
inds = nms_rotated_ext.nms_rotated(dets_th, scores, iou_thr)
RuntimeError: Not compiled with GPU support

---

而且在这种情况下运行get_map.py却也是有效的！

不断查看readme里出现的常见问题，github issue以及博客，得到博主回复：需重新编译nms_rotated

---

后安装vs2019

根据原博主的另一篇博客

安装好后进行编译，报如下错误：

win11下报错：fatal error C1083: 无法打开包括文件: “THC/THC.h”: No such file or directory
error: command 'C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.3\\bin\\nvcc.exe' failed with exit code 2

然后在issue里看到有人复现过程中出现了同样问题

后新建环境进行编译，编译成功！（总而言之：跟着博主的步骤来就好）扫雷了属于是！！！

---

至此继续运行train.py训练10/100epoch后Get map成功（阶段性胜利....）

 新报错如下：

---

 

2023年4月4日22:01:34

---