waste-classifier / requirements.txt
santit96's picture
Create the streamlit app that classifies the trash in an image into classes
fa84113
raw
history blame
355 Bytes
albumentations>=0.5.2
efficientnet_pytorch
fastai==2.7.13
funcy==1.15
iterative-stratification==0.1.6
matplotlib==3.8.2
numpy==1.26.2
omegaconf>=2.0
opencv-python==4.8.1.78
opencv-python-headless==4.8.1.78
pycocotools>=2.0.0
pytorch_lightning
pyyaml
rembg==2.0.53
scikit-learn==1.3.2
scikit-plot
scipy==1.11.4
streamlit
timm
torch
torchvision
tqdm==4.66.1