Skip to content

Instantly share code, notes, and snippets.

View PrimadonnaGit's full-sized avatar
🏋️‍♂️

JinWooHyun PrimadonnaGit

🏋️‍♂️
View GitHub Profile
@PrimadonnaGit
PrimadonnaGit / ssd detector_datasets.py
Created September 3, 2019 11:58
SSD detector for pytorch
import torch
from torchvision.utils.data import Dataset
import json
import os
from PIL import Image
from utils import transform
class PascalVOCDataset(Dataset):
"""
A Pytorch Dataset class to be used in a PyTorch DataLoader to create batches.