torchquantum icon indicating copy to clipboard operation
torchquantum copied to clipboard

What changes need make for quantum RGB image?

Open Wing90909090 opened this issue 2 years ago • 1 comments

Here is code, https://github.com/mit-han-lab/torchquantum/blob/main/examples/quanvolution/quanvolution_trainable_quantum_layer.py What need make here to read 128x128 RGB image?

Thanks you

Wing90909090 avatar Jun 13 '23 10:06 Wing90909090

Hi! Sorry about getting to this so late. I would imagine the main change would be updating the DataLoader to be able to process the extra channels. There probably will also be some minor updates to the models to account for the different shapes, but I think the biggest first step would be updating the DataLoader.

I’ve been meaning to write an example of this, but I haven’t had the bandwidth. I’d be happy to help as you work through this, so please reach out if you have any questions! And, once you’re done, feel free to contribute with the example!