pip install git+https://github.com/exla-ai/InferX.git

from inferx.models.resnet34 import ResNet34
import cv2

# Initialize the model (automatically detects your hardware)
model = ResNet34()

# Load and classify an image
image = cv2.imread("path/to/your/image.jpg")
results = model.inference(image)

# Print top predictions
for prediction in results['predictions'][:5]:
    print(f"{prediction['class']}: {prediction['confidence']:.3f}")

from inferx.models.resnet34 import ResNet34
import cv2
import os

model = ResNet34()

# Process multiple images
image_folder = "path/to/images/"
images = []
filenames = []

for filename in os.listdir(image_folder):
    if filename.endswith(('.jpg', '.png', '.jpeg')):
        image_path = os.path.join(image_folder, filename)
        image = cv2.imread(image_path)
        images.append(image)
        filenames.append(filename)

# Batch inference
results = model.batch_inference(images)

# Process results
for filename, result in zip(filenames, results):
    top_prediction = result['predictions'][0]
    print(f"{filename}: {top_prediction['class']} ({top_prediction['confidence']:.3f})")

from inferx.models.resnet34 import ResNet34
import cv2

model = ResNet34()

# Open camera
cap = cv2.VideoCapture(0)

while True:
    ret, frame = cap.read()
    if not ret:
        break
    
    # Run inference
    results = model.inference(frame)
    
    # Display top prediction
    if results['predictions']:
        top_pred = results['predictions'][0]
        text = f"{top_pred['class']}: {top_pred['confidence']:.2f}"
        
        cv2.putText(frame, text, (10, 30), 
                   cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
    
    cv2.imshow('ResNet34 Classification', frame)
    
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

# Filter predictions by confidence threshold
results = model.inference(
    image, 
    confidence_threshold=0.5,
    top_k=10
)

# Only show high-confidence predictions
high_confidence = [
    pred for pred in results['predictions'] 
    if pred['confidence'] > 0.7
]

for pred in high_confidence:
    print(f"High confidence: {pred['class']} ({pred['confidence']:.3f})")

{
    'predictions': [
        {
            'class': str,        # Class name (e.g., "golden retriever")
            'class_id': int,     # ImageNet class ID
            'confidence': float  # Confidence score (0-1)
        }
    ],
    'inference_time': float,     # Time taken for inference (seconds)
    'preprocessing_time': float  # Time taken for preprocessing (seconds)
}

from inferx.models.resnet34 import ResNet34
import os
import shutil

model = ResNet34()

# Organize photos by detected objects
source_folder = "unsorted_photos/"
organized_folder = "organized_photos/"

for filename in os.listdir(source_folder):
    if filename.endswith(('.jpg', '.png', '.jpeg')):
        image_path = os.path.join(source_folder, filename)
        image = cv2.imread(image_path)
        
        results = model.inference(image)
        
        if results['predictions']:
            top_class = results['predictions'][0]['class']
            
            # Create category folder
            category_folder = os.path.join(organized_folder, top_class)
            os.makedirs(category_folder, exist_ok=True)
            
            # Move file
            destination = os.path.join(category_folder, filename)
            shutil.move(image_path, destination)
            
            print(f"Moved {filename} to {top_class} folder")

# Example for manufacturing quality control
def check_product_quality(image_path, expected_class="product_name"):
    model = ResNet34()
    image = cv2.imread(image_path)
    
    results = model.inference(image, top_k=5)
    
    # Check if expected class is in top predictions
    for pred in results['predictions']:
        if expected_class.lower() in pred['class'].lower():
            if pred['confidence'] > 0.8:
                return "PASS", pred['confidence']
            else:
                return "LOW_CONFIDENCE", pred['confidence']
    
    return "FAIL", 0.0

# Test product
status, confidence = check_product_quality("product_image.jpg", "bottle")
print(f"Quality Check: {status} (Confidence: {confidence:.3f})")

# Basic content filtering example
def classify_image_content(image_path):
    model = ResNet34()
    image = cv2.imread(image_path)
    
    results = model.inference(image, top_k=10)
    
    # Define categories of interest
    animal_classes = ['dog', 'cat', 'bird', 'horse', 'cow']
    vehicle_classes = ['car', 'truck', 'motorcycle', 'bicycle']
    nature_classes = ['tree', 'flower', 'mountain', 'beach']
    
    categories = []
    
    for pred in results['predictions']:
        class_name = pred['class'].lower()
        
        if any(animal in class_name for animal in animal_classes):
            categories.append(f"Animal: {pred['class']}")
        elif any(vehicle in class_name for vehicle in vehicle_classes):
            categories.append(f"Vehicle: {pred['class']}")
        elif any(nature in class_name for nature in nature_classes):
            categories.append(f"Nature: {pred['class']}")
    
    return categories

categories = classify_image_content("test_image.jpg")
print("Image contains:", categories)

✨ InferX - ResNet34 Model ✨
🔍 Device Detected: AGX_ORIN
⠏ [0.5s] Loading ResNet34 model
✓ [0.7s] Ready for image classification