Skip to content

train过程出错 #15

@clearliu777

Description

@clearliu777

File "D:/untitled2/Segmentation/unet_mobile/train.py", line 42, in generate_arrays_from_file
one_hot_label = np.eye(NCLASSES)[np.array(label, np.int16)]
IndexError: index 128 is out of bounds for axis 0 with size 2

做的是2分类 但是这个地方显示128是为什么呀 求up主指导

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions