Skip to content

Instantly share code, notes, and snippets.

View Bilaerl's full-sized avatar

Bilal Babale Bilaerl

View GitHub Profile
@Bilaerl
Bilaerl / ewdism_dataset.py
Created October 28, 2023 09:53
Code Sample to share with Stickermule
import numpy as np
import torch
from torchvision.io import read_image
from torch.utils.data import Dataset
from pycocotools.coco import COCO
class EwdismDataset(Dataset):
""" Ewdism Dataset