Commit Graph

9 Commits

Author SHA1 Message Date
evazion aaaae88e77 autotag: fix bug when reading file from stdin.
Another attempt at fixing #2. b216057 had a bug where you couldn't pass
a file through stdin because we tried to read the file twice, which we
can't do if the input is a pipe.
2022-06-29 20:02:18 -05:00
evazion b2160576fc Better fix for #2 (Crashes when encountering files that aren't images)
* Only ignore invalid files in the `autotag` script, not in the
  web service. In the web service we want to return an error if given an
  invalid or corrupt file.

* Use the logger to log a warning instead of printing directly to stderr.
2022-06-29 18:25:11 -05:00
evazion 7c4f52f7d5 Revert "Print correct filenames when files are skipped"
This reverts commit b0b93eb471.

Broke the webapp output.
2022-06-29 14:47:29 -05:00
dokutan b0b93eb471 Print correct filenames when files are skipped 2022-06-28 19:01:43 +02:00
dokutan 9fd7697c80 Print skipped files 2022-06-28 18:44:49 +02:00
dokutan 8e4690b1b3 Skip files that aren't images 2022-06-26 21:05:10 +02:00
evazion 96b6a12924 autotag: accept files from stdin with '-'.
Fix `autotag` so you can pass a filename of '-' to read a file from
stdin. This way you can do this:

  docker run --rm -i ghcr.io/danbooru/autotagger autotag - < image.jpg

...to perform prediction on a single file outside of Docker.
2022-06-22 15:25:48 -05:00
evazion 07b84d73fe Do batch prediction.
Do batch prediction inside the `predict` method instead of calling
`predict` once for each image.
2022-06-21 20:08:49 -05:00
evazion 55d343cd3d Initial commit. 2022-06-20 23:33:54 -05:00