Instructions to use lrzjason/noise-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use lrzjason/noise-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="lrzjason/noise-classifier") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("lrzjason/noise-classifier") model = AutoModelForImageClassification.from_pretrained("lrzjason/noise-classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
-
Update README.md
Verified signature18e84b9 -
Update README.md
Verified signatureAn vit classifier for handling noise image like this  It has limitation inbetween clear and noise ``` from datasets import load_dataset from PIL import Image from transformers import ViTImageProcessor, ViTForImageClassification, TrainingArguments, Trainer import torch import numpy as np from datasets import load_metric import os import shutil model_name_or_path = 'lrzjason/noise-classifier' image_processor = ViTImageProcessor.from_pretrained(model_name_or_path) model = ViTForImageClassification.from_pretrained(model_name_or_path) input_dir = '' file = 'b5b457f4-5b52-4d68-be1b-9a2f557465f6.jpg' image = Image.open(os.path.join(input_dir, file)) inputs = image_processor(image, return_tensors="pt") with torch.no_grad(): logits = model(**inputs).logits # model predicts one of the 1000 ImageNet classes predicted_label = logits.argmax…94a42b3 -
Upload 8 files
Verified signatureInitf324401 -
initial commit
Verified signaturea2f7983