Skip to content

Commit

Permalink
lstm.train: allow .box from .raw.png too
Browse files Browse the repository at this point in the history
  • Loading branch information
bertsky authored and stweil committed Nov 12, 2022
1 parent 78bcc0d commit 8f4aae7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/ccstruct/boxread.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ static std::string BoxFileName(const char *image_filename) {
std::string box_filename = image_filename;
size_t length = box_filename.length();
std::string last = (length > 8) ? box_filename.substr(length - 8) : "";
if (last == ".bin.png" || last == ".nrm.png") {
if (last == ".bin.png" || last == ".nrm.png" || last == ".raw.png") {
box_filename.resize(length - 8);
} else {
size_t lastdot = box_filename.find_last_of('.');
Expand Down

0 comments on commit 8f4aae7

Please sign in to comment.