Git Product home page Git Product logo

dockerpytorchopencv's Introduction

Docker-PyTorch-OpenCV

Dockerfile for an Image installed with OpenCV and Pytorch for a hassle-free development environment.

Steps to Build the Image from the Dockerfile :

  • Clone or download the repository on your local machine: git clone https://github.com/BuffetCodes/DockerPyTorchOpenCV.git
  • Navigate to the cloned repo using: cd DockerPyTorchOpenCV
  • Build the image using: sudo docker build . (do not forget the period after build !)
  • Once the image has been created, run it using: sudo docker run -it --rm <image_ID>
  • The command line interface of the container can then be used for working on OpenCV and Pytorch related projects.

Note: The image can directly be downloaded from dockerhub using sudo docker pull himanshus16100/cv_pytorch. The final image size is 2.78 GB.

dockerpytorchopencv's People

Contributors

himanshusinghgh avatar krshrimali avatar

Stargazers

 avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

Forkers

farandal

dockerpytorchopencv's Issues

Add Libtorch and tests to the Dockerfile

  • Add opencv_contrib to the Dockerfile.
  • Revise cmake commands and add flags.
  • Add Libtorch to the Dockerfile.
  • Add sample test codes to test the libraries installed. Assert that they should pass.

Include tests for PyTorch and OpenCV in C++

Just like we have tests for checking PyTorch and OpenCV imports in Python (see below), it will be great to have tests for these libraries in C++.

python3 -c "\
try:
    import cv2
    print('OpenCV was installed successfully')
except ImportError:
    print('Error occured, file issue on GitHub repo. OpenCV not installed successfully')"

python3 -c "\
try:
    import torch
    print('Torch was installed successfully')
except ImportError:
    print('Error occured, file issue on GitHub repo. PyTorch installed successfully')"
  • Add tests for PyTorch in C++
  • Add tests for OpenCV in C++

Update: No need to add tests for Libtorch since we explicitly link to CMake config files using CMAKE_PREFIX_PATH flag.

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.