| # ALSATIX ResNet50 Model |
|
|
| This model is trained to classify images into 5 categories: |
|
|
| 1. **Alkol**: Alcohol-related images |
| 2. **Normal**: Regular images |
| 3. **NSFW**: Not Safe for Work images |
| 4. **Silah**: Weapon-related images |
| 5. **Tutun**: Tobacco-related images |
|
|
| ## Model Architecture |
| - Base: ResNet50 pre-trained on ImageNet |
| - Custom top layers: Dense (256 units), Dropout (0.5), Output (5 classes) |
|
|
| ## Usage |
|
|
| To use this model for image classification: |
|
|
| ```python |
| from transformers import TFAutoModelForImageClassification, AutoImageProcessor |
| |
| model = TFAutoModelForImageClassification.from_pretrained("iammbrn/alsatix_image_control_model") |
| processor = AutoImageProcessor.from_pretrained("iammbrn/alsatix_image_control_model") |
| |
| # Preprocess your image |
| image = processor(image, return_tensors="pt") |
| predictions = model(**image) |
| |