# Image-Classification-Framework

**Repository Path**: code_killer/Image-Classification-Framework

## Basic Information

- **Project Name**: Image-Classification-Framework
- **Description**: 一个简单的的猫狗图像分类框架(基于pytorch实现)，该框架经过简单的修改就可以在自己的数据集上运行或用于进行其他深度学习任务。
- **Primary Language**: Python
- **License**: MIT
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2022-06-04
- **Last Updated**: 2025-04-06

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Image-Classification-Framework
This is a simple and useful image classification framework based on pytorch. You can easily modify it for other deep learning tasks.

# Requirements
This code are tested under python 3.8.0, torch 1.8.0 and torchvision 0.9.0.
Other dependencies are listed in requirements.txt. You can install them with the following commands:

    pip install -r requirements.txt


# Datasets
Dataset link: https://www.kaggle.com/datasets/googolples/cat-dog-images

If you want to run this code on custom dataset, you just need to set the root of your data in --data_root and organize your data like the example dataset.


# Run

    python main.py