Skip to content

Navifra-Kerry/transfer-learning-Libtorch

Repository files navigation

LibTorch Example of transfer-learning

https://blog.kerrycho.com/TransferLearning-Libtorch/

GitHub Actions status

Example

GitHub Actions status

Requirements

  • OpenCV 4.1.1
  • Libtorch 1.4
  • CUDA 10.2
  • Visual studio 2019
  • Windows 10

Install with Scripts

transfer-learning
 - Scripts
    -install.bat
    
Run install.bat
It will download and install automatically.

GitHub Actions status

Convert Python Model to C++

transfer-learning
 - Model
    convert.py
Run Python Script