yolov7 icon indicating copy to clipboard operation
yolov7 copied to clipboard

Re-parameterization.ipynb >> 'Model' object has no attribute 'nc'

Open HeungJunKim opened this issue 1 year ago • 1 comments

I trained 300 epochs and chose the last model.

model : YOLOv7-X weight file name : best_299.pt

but, An error occurs at this location.

in Re-parameterization.ipynb model.nc = ckpt['model'].nc 스크린샷, 2022-07-27 18-21-48 스크린샷, 2022-07-27 18-09-54

If that line is omitted, the conversion is performed successfully, but the mAP is slightly reduced.

HeungJunKim avatar Jul 27 '22 09:07 HeungJunKim

Just set the value of model.nc as your training number class.

Eliza-and-black avatar Aug 08 '22 11:08 Eliza-and-black