PavlosMelissinos / enet-keras

A keras implementation of ENet (abandoned for the foreseeable future)
MIT License
115 stars 46 forks source link

Add support for more datasets #12

Open PavlosMelissinos opened 7 years ago

PavlosMelissinos commented 7 years ago

CamVid, Cityscapes or SUN, which should help with #11 (cityscapes is the most recent one).

cocostuff, as a first stress test for #8

PavlosMelissinos commented 7 years ago

Progress report:

Still working on multiple datasets, at the moment I'm still in the process of polishing the dataset API (check out the datasets_overhaul branch for progress on that front).

Thing is, that branch and the whole dataset API thing should really not be a part of this repo, that's why I've not merged yet. However, the code works and I think some of my design ideas are decent so I've also opened an issue on keras-contrib to discuss the API. Eventually, keras does not have to be a requirement but it's a start.