UNet partial report#190

Open

recoverer214 wants to merge 6 commits intoshakes76:mainfrom

recoverer214:main

Commits on Dec 22, 2024

UNet structure is complete.
recoverer214
committed
Imported image processing files from the project. Implemented custom DataLoader.
recoverer214
committed
Added train validate test loop for each epoch.
recoverer214
committed
Don't modify the shape of tensors here (Rather in train initialization).
recoverer214
committed
Modified loss function from BCE to Cross entropy. Omitted unnecessary main loop. Modified test loop. Added images to Readme file.
recoverer214
committed

Commits on Dec 23, 2024

Fixed dependence of test file
recoverer214
committed