search

speedinghzl / pytorch-segmentation-toolbox

PyTorch Implementations for DeeplabV3 and PSPNet
MIT License
768 stars 167 forks source link

Custom dataset #24

Closed Jorisfournel closed 5 years ago

Jorisfournel commented 5 years ago

Hi, Thanks for the contribution, could you show us what part of your code is to be modified in order to train on a custom dataset. Also in order to implement custom loss functions ? Thanks in advance, Joris

speedinghzl commented 5 years ago

In fact, the organization of this repo is clear and easy-to-read. The data loader is defined in file dataset.py and loss functions are defined in criterion.py.