File size: 841 Bytes
8988d66 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# Custom CNN Model for Image Classification
This repository contains a custom ResNet-like CNN trained for image classification. The model is provided in both PyTorch (`pytorch_model.bin`) and Safetensors (`model.safetensors`) formats for easy inference and transfer.
## Usage
- Load model weights and architecture from Hugging Face Hub.
- Preprocess images using the provided mean/std and img_size in `preprocessor_config.json`.
- See `train.py` for full training and inference pipeline.
## Files
- `pytorch_model.bin`: PyTorch weights
- `model.safetensors`: Safetensors weights
- `config.json`: Model config
- `preprocessor_config.json`: Preprocessing config
- `train.py`: Source code
- `prediction.csv`: Example predictions
## Citation
If you use this model, please cite the repository: https://huggingface.co/DungND1107/IntroAIver2
|