Skip to content

Commit 73902c6

Browse files
committed
Fix bug with img_folder_dataset not correctly saving the cache
1 parent e1c9116 commit 73902c6

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

data/img_folder_dataset.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@ import (
2121
)
2222

2323
type imgMetadata struct {
24-
filepath string
25-
category int
24+
Filepath string
25+
Category int
2626
}
2727

2828
type ImgFolderDataset struct {
@@ -235,8 +235,8 @@ func (d *ImgFolderDataset) readFileNames() error {
235235
d.Count++
236236

237237
d.images = append(d.images, imgMetadata{
238-
filepath: imgFilePath,
239-
category: categoryInt,
238+
Filepath: imgFilePath,
239+
Category: categoryInt,
240240
})
241241

242242
now := time.Now().Unix()
@@ -325,7 +325,7 @@ func (d *ImgFolderDataset) getRow() (string, int, error) {
325325
}
326326
img := d.images[d.generatorOffset]
327327
d.generatorOffset++
328-
return img.filepath, img.category, nil
328+
return img.Filepath, img.Category, nil
329329
} else {
330330
panic("Non shuffled mode not implemented")
331331
}

readme.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ See `ideas-todo.md` for what's in store
2626

2727
Versions starting with v0 are liable to change radically.
2828

29-
- Tensorflow 2.6 experimental support: `go get github.com/codingbeard/tfkg v0.2.6.19`
29+
- Tensorflow 2.6 experimental support: `go get github.com/codingbeard/tfkg v0.2.6.20`
3030

3131
## Requirements
3232

0 commit comments

Comments
 (0)