Views
No views yet
LibreCLIP. No
training and no fixed label set: call set_classes([...]), then predict.ViT-B-32, pretrained tag laion2b_s34b_b79k).
Copyright (c) 2021 OpenAI; (c) 2012-2021 OpenCLIP authors. Licensed under the
MIT License.NOTICE.weights/convert_clip_weights.py in the
LibreYOLO source repository.1from libreyolo import LibreCLIP
2
3model = LibreCLIP("LibreCLIPb32-cls.pt") # autodownloads from this repo
4model.set_classes(["a forklift", "an empty aisle", "a spill"])
5r = model.predict("warehouse.jpg")[0]
6print(model.names[r.probs.top1], float(r.probs.top1conf))