Object Classifier
Reason this release was yanked:
Upgrade objifier==0.1.1 that support yaml
Project description
OBJECT CLASSIFIER
Clone this repository
- for SSH
git clone git@github.com:1chimaruGin/Object-classifier.git
- for https
https://github.com/1chimaruGin/Object-classifier.git
Requirements
pip install -U requirements.txt
Dataset
- the dataset directory should be the following format.
# for example, dog vs cat classification
data/
-train/
dog/
-*.jpg or *.png
cat/
-*.jpg or *.png
-val/
dog/
-*.jpg or *.png
cat/
-*.jpg or *.png
Usage
cd Object-classifier
-
Update number of classes and names in data.yaml
-
For training model(ResNet)
$ python main.py -m [mode: train] -opt [optimizer: (default='SGD')] -epochs [epochs: (default=25)]
- For training model(EfficientNet)
$ python main.py -m [mode: train] -opt [optimizer] -epochs [epochs] -backbone [backbone: efficientNet] -lvl [efficientNet level]
- For prediction
$ python main.py -m [mode: predict] -im [input image] - backbone [backbone: ResNet or efficientNe] -lvl [efficientNet level]
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
objifier-0.1.0.tar.gz
(20.5 kB
view hashes)
Built Distribution
objifier-0.1.0-py3-none-any.whl
(23.0 kB
view hashes)