| 178 | } |
| 179 | |
| 180 | int main(int argc, char** argv) |
| 181 | { |
| 182 | int32_t width{1282}; |
| 183 | int32_t height{1026}; |
| 184 | |
| 185 | SampleSegmentation sample("fcn-resnet101.engine"); |
| 186 | |
| 187 | gLogInfo << "Running TensorRT inference for FCN-ResNet101" << std::endl; |
| 188 | if (!sample.infer("input.ppm", width, height, "output.ppm")) |
| 189 | { |
| 190 | return -1; |
| 191 | } |
| 192 | |
| 193 | return 0; |
| 194 | } |