Machine-Learning-Collection icon indicating copy to clipboard operation
Machine-Learning-Collection copied to clipboard

add header=None to pd.read_csv

Open samrere opened this issue 2 years ago • 0 comments

Hi just wanna report a small mistake in YOLO dataset.py

 class VOCDataset(...):
    def __init__(...):
        self.annotations = pd.read_csv(csv_file)

should be self.annotations = pd.read_csv(csv_file, header=None), otherwise the first line of the file is used as header and thus not used in training/testing

samrere avatar Mar 26 '23 16:03 samrere