Fixed np.int issue in u7 branch and under the seg/utils/dataloader.py

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

GNU General Public License v3.0

13.4k stars 4.23k forks source link

Fixed np.int issue in u7 branch and under the seg/utils/dataloader.py #2080

Open Nomaanrizvi opened 2 months ago

Nomaanrizvi commented 2 months ago

corrected model training bug by changing the data-type of the batch index(line 488)

This error occurs while training a custom dataset using the YOLOv7 segmentation module in a Colab environment. The issue is related to the numpy library, specifically an outdated use of np.int, which was deprecated starting in NumPy version 1.20.