https://github.com/ajithvcoder/Custom_Object_Classification
Simple Custom object classification with pytorch | Custom Model | Custom dataset | ONNX inference
https://github.com/ajithvcoder/Custom_Object_Classification
classification custom customdataset deep-learning onnx pytorch simple
Last synced: about 1 month ago
JSON representation
Simple Custom object classification with pytorch | Custom Model | Custom dataset | ONNX inference
- Host: GitHub
- URL: https://github.com/ajithvcoder/Custom_Object_Classification
- Owner: ajithvcoder
- Created: 2023-05-31T11:24:06.000Z (almost 2 years ago)
- Default Branch: main
- Last Pushed: 2023-05-31T14:07:20.000Z (almost 2 years ago)
- Last Synced: 2024-05-22T03:07:32.933Z (about 1 year ago)
- Topics: classification, custom, customdataset, deep-learning, onnx, pytorch, simple
- Language: Python
- Homepage:
- Size: 4.42 MB
- Stars: 3
- Watchers: 3
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Simple Custom_Object_Classification with pytorch
Simple Custom object classification with pytorch | Custom Model | Custom dataset | ONNX inference
- Genric Data preparation
- Installation
- Training
- Testing
- Onnx model inference
**Gallery**
### Genric Data preparation
- Prepare the dataset like below tree
```
- data
- Fruits (dataset name)
- train
- class 1
- class 2
- ...
- class n
- val
- class 1
- class 2
- ...
- class n
- test
- class 1
- class 2
- ...
- class n
```
- I have took fruits dataset from kaggle for example purpose [link here](https://www.kaggle.com/datasets/shreyapmaher/fruits-dataset-images)
- change the path name in "train_dir" and "val_dir" in main.py
### Installation
- pip install -r requirements.txt
### Training
- python main.py
### Testing
Make sure if you have changed weights filename, train folder, input image in predict.py
- python predict.py
### Exporting
Make sure if you have changed the weights filename in export.py
- python export.py
### Onnx model inference
Make sure if you have changed weights filename, train folder, input image in predict.py
- python onnx_inference.py
Kindly raise issues or mail me if you have any questions
Medium blog - [here](https://medium.com/@ajithkumarv/8a19e5ca5e27)
**Credits**
- [Custom-CNN-based-Image-Classification-in-PyTorch](https://github.com/developer0hye/Custom-CNN-based-Image-Classification-in-PyTorch)
- [Creating custom dataloader](https://medium.com/analytics-vidhya/creating-a-custom-dataset-and-dataloader-in-pytorch-76f210a1df5d)