You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
@goldwater668 labels.txt file is used in vision/datasets/voc_dataset.py.
below is the snippet of code : which will check your custom label.txt is available that make classes according to that else take "face" and "Background" as default.
# if the labels file exists, read in the class names
label_file_name = self.root / "labels.txt"
if os.path.isfile(label_file_name):
class_string = ""
with open(label_file_name, 'r') as infile:
for line in infile:
class_string += line.rstrip()
# classes should be a comma separated list
classes = class_string.split(',')
# prepend BACKGROUND as first class
classes.insert(0, 'BACKGROUND')
classes = [elem.replace(" ", "") for elem in classes]
self.class_names = tuple(classes)
logging.info("VOC Labels read from file: " + str(self.class_names))
Want to know where the \data\retinaface_labels\train\labels.txt files used in the author's training are generated?
The text was updated successfully, but these errors were encountered: