Commit 87bc664b authored by Davis King's avatar Davis King

Changed the XML format output by net_to_xml(). Specifically, the XML tag for

affine layers was changed so to use the same conventions as other layers that
support convolutional vs fully connected modes.
parent 917ea27b
......@@ -1595,14 +1595,17 @@ namespace dlib
friend void to_xml(const affine_& item, std::ostream& out)
{
out << "<affine";
if (item.mode==CONV_MODE)
out << " mode='conv'";
out << "<affine_con>\n";
else
out << " mode='fc'";
out << ">\n";
out << "<affine_fc>\n";
out << mat(item.params);
out << "</affine>\n";
if (item.mode==CONV_MODE)
out << "</affine_con>\n";
else
out << "</affine_fc>\n";
}
private:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment