a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Improve ignore_first logic

Browse Source
This commit is contained in:
James Betker 2020-11-03 11:56:32 -07:00
parent 658a267bab
commit e990be0449
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/data/chunk_with_reference.py
View File

@ -10,10 +10,7 @@ class ChunkWithReference:
self.tiles, _ = util.get_image_paths('img', self.path) self.tiles, _ = util.get_image_paths('img', self.path)
self.strict = opt['strict'] if 'strict' in opt.keys() else True self.strict = opt['strict'] if 'strict' in opt.keys() else True
if 'ignore_first' in opt.keys(): if 'ignore_first' in opt.keys():
self.ignore = opt['ignore_first'] self.tiles = self.tiles[opt['ignore_first']:]
self.tiles = self.tiles[self.ignore:]
else:
self.ignore = 0
# Odd failures occur at times. Rather than crashing, report the error and just return zeros. # Odd failures occur at times. Rather than crashing, report the error and just return zeros.
def read_image_or_get_zero(self, img_path): def read_image_or_get_zero(self, img_path):