Spaces:
Sleeping
Sleeping
File size: 322 Bytes
fa84113 |
1 2 3 4 5 6 7 8 9 |
CLAS_FILEPATH = "models/resnet50-classifier.pkl"
DET_FILEPATH = "models/efficientdet-d2-detector.pth.tar"
CLASSES = ["cardboard", "compost", "glass", "metal", "paper", "plastic", "trash"]
DET_NAME = "tf_efficientdet_d2"
CLAS_THRESHOLD = 0.5
DET_THRESHOLD = 0.17
DEVICE = "cpu"
OUTPUT_IMG_FILEPATH = "classified_image.jpg"
|