search

Xiyue-Wang / TransPath

GNU General Public License v3.0
248 stars 32 forks source link

patch classification #19

Open Rukhmini opened 1 year ago

Rukhmini commented 1 year ago

Hello, How does your pipeline work for patch classification? After extracting the features from the patches, for classification what code needs to be executed? TIA.

Xiyue-Wang commented 1 year ago

Hi, It's very easy. It's a common sense in CV. I've written the loading process in ctrans_lincls.py. For the linear classification training process, you can refer to Moco v2/v3(https://github.com/facebookresearch/moco-v3,https://github.com/facebookresearch/moco)