yukun-hh/trash-division
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

修改dataloader

Browse source
This commit is contained in:
yukun-hh 2026-04-12 20:10:53 +08:00
parent af1bed54ea
commit e380094fa1
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
Dataloader.py
View file

@ -13,7 +13,7 @@ import os
from PIL import Image from PIL import Image
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import numpy as np import numpy as np
import pandas as pd
def create_dataloaders(data_root='..', def create_dataloaders(data_root='..',
batch_size=32, batch_size=32,