https://github.com/ternaus/check_orientation
Model to check if image was rotated by 90, 180, 270 degrees.
https://github.com/ternaus/check_orientation
computer-vision deep-learning image-classification orientation-detection
Last synced: 12 months ago
JSON representation
Model to check if image was rotated by 90, 180, 270 degrees.
- Host: GitHub
- URL: https://github.com/ternaus/check_orientation
- Owner: ternaus
- License: apache-2.0
- Created: 2020-11-05T19:32:41.000Z (over 5 years ago)
- Default Branch: main
- Last Pushed: 2020-11-19T00:35:41.000Z (over 5 years ago)
- Last Synced: 2024-12-28T14:26:56.342Z (over 1 year ago)
- Topics: computer-vision, deep-learning, image-classification, orientation-detection
- Language: Python
- Homepage:
- Size: 50.8 KB
- Stars: 93
- Watchers: 3
- Forks: 17
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- Funding: .github/FUNDING.yml
- License: LICENSE
Awesome Lists containing this project
README
# Check orientation
Models to check if image was rotated by 0, 90, 180, 270 degrees.
## Installation
`pip install -U check_orientation`
### Example inference
Colab notebook with the example: [](https://colab.research.google.com/drive/1HIGM_b8rH20N414ROZB0HN9w5p4ko2Bd?usp=sharing)
## Training
### Define the config.
Example at [check_orientation/configs](check_orientation/configs)
### Define the environmental variable `TRAIN_IMAGE_PATH` that points to the folder with train dataset.
Example:
```bash
export TRAIN_IMAGE_PATH=
```
### Define the environmental variable `VAL_IMAGE_PATH` that points to the folder with validation dataset.
Example:
```bash
export VAL_IMAGE_PATH=
```
### Training
```
python -m check_orientation.train -c
```
### Inference
```bash
python -m torch.distributed.launch --nproc_per_node= check_orientation/inference.py \
-i \
-c \
-w \
-o \
--fp16
```
### Pre-trained models
Models were pre-trained on the [OpenImages dataset](https://storage.googleapis.com/openimages/web/index.html).
| Models | Validation accuracy | Config file | Weights |
| ------------- |:--------------------:| :------------:| :------: |
| swsl_resnext50_32x4d|0.9128| [Link](check_orientation/configs/2020-11-16.yaml)| [Link](https://github.com/ternaus/check_orientation/releases/download/v0.0.3/2020-11-16_resnext50_32x4d.zip)|