About the multiple categories in code

Open alimagic opened this issue 5 years ago • 1 comments

Dear Author: When I use this code to train for multiple categories, I find the code in "dataset.py" about line 37 "self._valid_ids = [1]" .Does it need to be modified? For example, two categories, self._valid_ids = [1, 2].

Dec 11 '20 16:12 alimagic

Yes. I forgot to emphasize that. Sorry.

Dec 12 '20 16:12 ZeroE04