search

danfenghong / IEEE_TGRS_SpectralFormer

Danfeng Hong, Zhu Han, Jing Yao, Lianru Gao, Bing Zhang, Antonio Plaza, Jocelyn Chanussot. Spectralformer: Rethinking hyperspectral image classification with transformers, IEEE Transactions on Geoscience and Remote Sensing (TGRS), 2021
224 stars 38 forks source link

关于训练部分的pt文件 #15

Open handsomezhuo opened 2 years ago

handsomezhuo commented 2 years ago

老师，您好，有幸使用了您的代码，在代码运行对测试集进行运行的时候，我发现了您使用的是已经保存好了的训练模型也就是VIT_indian.pt文件，如果说我想用我自己的数据集去运行代码的话，在训练部分是只需要将我数据集划分为TE,TR,label，但是在测试集部分就需要自己生成一个关于数据集在训练部分的参数，我想问一下您，我如何能在自己数据集上训练结束之后，生成相对应的测试pt文件呢

danfenghong commented 2 years ago

你需要shi用自己的数据在训练样本上训练出自己的模型，然后用于自己数据的测试。

handsomezhuo @.***> 于2022年8月9日周二 09:59写道：

老师，您好，有幸使用了您的代码，在代码运行对测试集进行运行的时候，我发现了您使用的是已经保存好了的训练模型也就是VIT_indian.pt文件，如果说我想用我自己的数据集去运行代码的话，在训练部分是只需要将我数据集划分为TE,TR,label，但是在测试集部分就需要自己生成一个关于数据集在训练部分的参数，我想问一下您，我如何能在自己数据集上训练结束之后，生成相对应的测试pt文件呢

— Reply to this email directly, view it on GitHub https://github.com/danfenghong/IEEE_TGRS_SpectralFormer/issues/15, or unsubscribe https://github.com/notifications/unsubscribe-auth/AFL2GZRPCPVLOARY2XT5IQ3VYG3QTANCNFSM557CGQGA . You are receiving this because you are subscribed to this thread.Message ID: @.***>

handsomezhuo commented 2 years ago

谢谢您，我已经可以使用我的数据集进行训练并且提取pt文件了

danfenghong commented 2 years ago

Good to know! :)

handsomezhuo @.***> 于2022年8月19日周五 10:11写道：

谢谢您，我已经可以使用我的数据集进行训练并且提取pt文件了

— Reply to this email directly, view it on GitHub https://github.com/danfenghong/IEEE_TGRS_SpectralFormer/issues/15#issuecomment-1220161266, or unsubscribe https://github.com/notifications/unsubscribe-auth/AFL2GZURAC7FD7WZ4RDUBNTVZ3UMHANCNFSM557CGQGA . You are receiving this because you commented.Message ID: @.***>