Skip to content

The pytvision package consists of my datasets, models, and image transformations methods for computer vision projects. This package also containing the synthetic render methods.

License

pedrodiamel/pytorchvision

Repository files navigation

PyTVision

Documentation Status License: MIT

The pytvision package consists of my datasets, models, and image transformations methods for computer vision projects. This package also containing the synthetic render methods. The backend transforms using opencv.

Requirements

pytorch 0.4.1
git clone https://github.com/pytorch/vision.git
cd vision
python setup.py install

Installation

From source:

python setup.py build_ext --inplace
python setup.py install

Docker

docker build -f "Dockerfile" -t pytv:latest .
docker run -ti --privileged --ipc=host --name pytv-dev -p 8080:8080 -p 6006:6006 -v $DATASETS:/.datasets pytv:latest /bin/bash

Dockercompose

docker-compose up --build -d
docker-compose down
docker exec -it pytv-dev /bin/bash

Books

Projects

Kaggle Projects

Documentation

The full documentation is available at doc

Dataset

  • cars169
  • cub2011
  • StanfordOnlineProducts
  • fer+

Models

  • inception_v4
  • nasnetalarge
  • dexpression
  • unet (unet11, unetresnet, ...)
  • pspnet

Transforms

  • Color transformation
  • Geometrical transformation
  • Blur + Noise transformation

Similar projects

About

The pytvision package consists of my datasets, models, and image transformations methods for computer vision projects. This package also containing the synthetic render methods.

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published