yolov7
yolov7 copied to clipboard
Re-parameterization.ipynb >> 'Model' object has no attribute 'nc'
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
If that line is omitted, the conversion is performed successfully, but the mAP is slightly reduced.
Just set the value of model.nc as your training number class.