DETR源码讲解：训练自己的数据集

斯麦璐k:
大佬，我在github的issue108里面找到了作者对num_class和nn.Linear的输出是num_class+1的解释： 假设我们有N个label，且label的下标是从1到N（中间没有空缺）。那必须将DETR网络的“num_classes”参数设置成N+1，而多出来的那一个id：N+1是用来预测“no_object”的。而到nn.Linear中的输出层的维度是num_classes+1是因为计算机编程语言的index默认是从0开始的，而为了使label的id和nn.Linear的输出的index对齐，所以还得再+1。如果数据集中的label的id没有使用0，那nn.Linear在index为0的这个维度的输出上就是dummy的，不需要考虑。同理，如果label中的id不是连续的，那nn.Linear对应的缺失的id是不会有预测结果的。

【回复】大佬，请问我的数据集只有两类，id是0，1.请问我的num_class应该是是多少啊
83124540675_bili:
up，求一个voc2json，predict，change_num_classes文件啦，谢谢

小鑫-鑫呀:
up很强 三联了 能求一份predict吗 [呲牙][呲牙][呲牙]

【回复】可以的话 detr segment也要一份 嘿嘿
就是想不到好的昵称:
predict文件是在里找呀 方便的话求一份[大哭]

【回复】回复 @奋斗在一线的吃货 : 这个应该是up自己写的
Sin花椰菜:
up，求predict的源码，谢谢啦！！！视频中的predict文件没有显示完整呢

【回复】回复 @Sin花椰菜 : z在哪里呀，没找到呢
【回复】找到了，在主页有。谢谢up[呲牙]
__yira__:
up，一键三连，求一份代码，跟着学习[星星眼]