Open ZoharKapach opened 4 years ago
Some of the things that need to be done to extend this implementation to multiple classes:
data.py
, the method __getitem__
of the class CSVDataset
, and the function csv_collator
, would need to be adapted accordingly.train.py
and locate.py
would also need modificationpaint.py
Could you specify the codes where the above changes has to be made? For example FCN we need to put n_classes=2.
Could you provide a list of required modifications in order for this to work for multiple classes?