Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career
!pip install jovian
Requirement already satisfied: jovian in /opt/conda/lib/python3.7/site-packages (0.2.18) Requirement already satisfied: requests in /opt/conda/lib/python3.7/site-packages (from jovian) (2.23.0) Requirement already satisfied: pyyaml in /opt/conda/lib/python3.7/site-packages (from jovian) (5.3.1) Requirement already satisfied: uuid in /opt/conda/lib/python3.7/site-packages (from jovian) (1.30) Requirement already satisfied: click in /opt/conda/lib/python3.7/site-packages (from jovian) (7.1.1) Requirement already satisfied: urllib3!=1.25.0,!=1.25.1,<1.26,>=1.21.1 in /opt/conda/lib/python3.7/site-packages (from requests->jovian) (1.24.3) Requirement already satisfied: idna<3,>=2.5 in /opt/conda/lib/python3.7/site-packages (from requests->jovian) (2.9) Requirement already satisfied: chardet<4,>=3.0.2 in /opt/conda/lib/python3.7/site-packages (from requests->jovian) (3.0.4) Requirement already satisfied: certifi>=2017.4.17 in /opt/conda/lib/python3.7/site-packages (from requests->jovian) (2020.6.20) WARNING: You are using pip version 20.2.2; however, version 20.2.3 is available. You should consider upgrading via the '/opt/conda/bin/python3.7 -m pip install --upgrade pip' command. 
%matplotlib inline

import os
import shutil
import random
import torch
import torchvision
import numpy as np
from sklearn import metrics
from sklearn.metrics import confusion_matrix
from PIL import Image
from matplotlib import pyplot as plt
import cv2
import jovian
torch.manual_seed(0)

print('Using PyTorch version', torch.__version__)
Using PyTorch version 1.5.1 
torch.cuda.is_available()
True
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
device
device(type='cuda')
train_dirs = {
    'clean': '../input/messy-vs-clean-room/images/train/clean',
    'messy': '../input/messy-vs-clean-room/images/train/messy',
}

test_dirs = {
  'clean': '../input/messy-vs-clean-room/images/val/clean',
    'messy': '../input/messy-vs-clean-room/images/val/messy',
}