pytorch_image_classifier - 副本
所属分类：Windows编程
开发工具：Python
文件大小：12KB
下载次数：0
上传日期：2021-03-19 00:36:56
上 传 者：
aaalphe
说明： 基于darknet模型，对昆虫进行识别的深度学习python代码，模型需到pytorch官网下载
(Based on the Darknet model, the deep learning Python code for insect recognition needs to be downloaded from the python official website)
文件列表:
pytorch_image_classifier (0, 2021-03-19)
pytorch_image_classifier\__pycache__ (0, 2020-12-06)
pytorch_image_classifier\__pycache__\global_config.cpython-36.pyc (255, 2020-12-06)
pytorch_image_classifier\data_loader (0, 2020-12-06)
pytorch_image_classifier\data_loader\__pycache__ (0, 2020-12-06)
pytorch_image_classifier\data_loader\__pycache__\data_loader.cpython-36.pyc (3227, 2020-12-06)
pytorch_image_classifier\data_loader\__pycache__\global_config.cpython-36.pyc (267, 2020-12-06)
pytorch_image_classifier\data_loader\data_loader.py (3634, 2020-12-06)
pytorch_image_classifier\data_loader\global_config.py (869, 2020-12-07)
pytorch_image_classifier\global_config.py (869, 2020-12-06)
pytorch_image_classifier\models (0, 2021-03-19)
pytorch_image_classifier\models\__pycache__ (0, 2020-12-06)
pytorch_image_classifier\models\__pycache__\fine_tune_model.cpython-36.pyc (505, 2020-12-06)
pytorch_image_classifier\models\fine_tune_model.py (1149, 2020-12-06)
pytorch_image_classifier\predict.py (2406, 2020-12-07)
pytorch_image_classifier\train.py (4747, 2020-12-07)
# PyTorch Image Classifier
## Updates
As for many users request, I released a new version of standared pytorch immage classification example at here: http://codes.strangeai.pro/aicodes_detail.html?id=30
It's more standared with **pytorch 1.0** support. It contains those features which is really useful to write a standared AI application:
- standared dataload to load data;
- separate net work defination and data process code (less couple);
- catch keyboard interrupt and resume training.
## Less Than 200 Line Train Codes and 25 Epochs, Got ***% Accuracy!
In this repo, I managed classify images into 2 kinds which is ants and
bees, but it's also very straightforward to train on more classes images.
The amazing thing is, using PyTorch, we can use less than 200 line code to get
a very hight accuracy of classify on images!!!
## Usage
To using this repo, some things you should to know:
* Compatible both of CPU and GPU, this code can automatically train on CPU or GPU;
* Models trained on GPU can also predict on CPU using predict.py;
* First run please run `bash download_datasets.sh` to obtain datasets;
* Model will be save after epochs;
* Image Size can be set in `global_config.py`.
## Future Work
This version if fine tune on ResNet18, in the future maybe implement some own network,
also fine tune on famous networks.
As well as more datasets.
## Copyright
This repo implement by Jin Fagang, and ofcourse PyTorch authors.
if you have any question, you can find me on wechat: `jintianiloveu`
近期下载者：
相关文件：
收藏者：