Skip to content

Instantly share code, notes, and snippets.

View biswajitcsecu's full-sized avatar

Biswajit Biswas biswajitcsecu

View GitHub Profile
@biswajitcsecu
biswajitcsecu / ImageClassificationTorchDemo.py
Last active April 30, 2021 01:35
ImageClassificationTorchDemo
from __future__ import print_function, division
import os
import torch
from torchvision import transforms, datasets
import pandas as pd
from skimage import io, transform
import numpy as np
import matplotlib.pyplot as plt
from torch.utils.data import Dataset, DataLoader
from torchvision import transforms, utils