How can I change the model to the one i use(eg. yolov5s

[WiselyC0430]

i noticed that the network in the demo code (4CH) supports the UNET and PHDNET, thus I wonder how can I change the network I am using into the code.

[Schobs]

@WiselyC0430 you can use any network for the results. Just make sure the error/uncertainty results are in the same format as described.