|
|
@@ -79,7 +79,7 @@ PYBIND11_MODULE(trtresnet, m){
|
|
|
.def_readwrite("class_label", &resnet::Attribute::class_label)
|
|
|
.def("__repr__", [](const resnet::Attribute &attr) {
|
|
|
std::ostringstream oss;
|
|
|
- oss << "Attribute(class_label: " << obj.class_label << ", confidence: " << obj.confidence << ")"
|
|
|
+ oss << "Attribute(class_label: " << attr.class_label << ", confidence: " << attr.confidence << ")"
|
|
|
return oss.str();
|
|
|
});;
|
|
|
|
