| 139 | } |
| 140 | |
| 141 | int main(int argc, char* argv[]) { |
| 142 | int device = argc > 1 ? atoi(argv[1]) : 0; |
| 143 | bool console = argc > 2 ? argv[2][0] == '-' : false; |
| 144 | |
| 145 | try { |
| 146 | af::setDevice(device); |
| 147 | af::info(); |
| 148 | |
| 149 | printf("Brain segmentation example\n"); |
| 150 | brain_seg(console); |
| 151 | |
| 152 | } catch (af::exception& e) { fprintf(stderr, "%s\n", e.what()); } |
| 153 | |
| 154 | return 0; |
| 155 | } |