Spaces:

rootstrap-org
/

waste-classifier

Sleeping

waste-classifier / constants.py

Create the streamlit app that classifies the trash in an image into classes

fa84113 almost 2 years ago

322 Bytes

	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"