andras/bipolaroidbipolaroid
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix "extract training"

Browse source
This commit is contained in:
Andras Schmelczer 2024-06-25 09:02:45 +01:00
parent 856dc83c77
commit 2475d7c8dd
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
3 changed files with 40 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/training/histogram_dataset.py
View file

@ -1,7 +1,7 @@
from torch.utils.data import Dataset
from typing import List, Optional, Tuple
from utils import compute_histogram
from .random_edit import random_edit
from operations.random_edit import random_edit
from PIL import Image
from tqdm import tqdm
import logging