/ic

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

ic

中文文档

This repository contains the tools and models (CNN, ViT and DKL Model) for image classification.

Usage

  1. Requirements:
    • Python >= 3.10
  2. Install PyTorch and torchvision following the official instructions.
  3. Install d2l:
    • Clone the d2l repository:
      git clone https://github.com/d2l-ai/d2l-en && cd d2l-en
    • Delete version specification of requirements in setup.py.
    • Install d2l using pip:
      pip install .
  4. Install other requirements:
    cd /path/to/ic/ && pip install -r requirements.txt
  5. Append the path of this repository in python scripts:
    import sys
    sys.path.append('/path/to/ic/')
  6. You can then import this package:
    import ic