search

CaptainEven / Vehicle-Car-detection-and-multilabel-classification

使用YOLO_v3_tiny和B-CNN实现街头车辆的检测和车辆属性的多标签识别 Using yolo_v3_tiny to do vehicle or car detection and attribute's multilabel classification or recognize
MIT License
596 stars 205 forks source link

关于多标签分类的问题 #13

Open konioyxgq opened 5 years ago

konioyxgq commented 5 years ago

你好，我看你在多标签分类中用到了B-CNN，在VehicleDC.py中这样写的： “X = torch.bmm(X, torch.transpose(X, 1, 2)) / (1 ** 2) # Bi-linear CNN” 我看了B-CNN的代码，并没有看到这样的方法。 请问你这样做的根据和目的是什么？ 也可能是我看漏掉了，还请作者指点。非常感谢

konioyxgq commented 5 years ago

@CaptainEven

CaptainEven commented 5 years ago

@konioyxgq 使用的是简化版的BCNN，两个分支共用一个网络，通过batch matrix multiply(torch.bmm)实现矩阵外乘。

konioyxgq commented 5 years ago

@CaptainEven 谢谢你的回复，你的BCNN的文章链接好像错了，因该是http://vis-www.cs.umass.edu/bcnn/docs/bcnn_iccv15.pdf 而不是https://arxiv.org/pdf/1709.09890.pdf

CaptainEven commented 5 years ago

@konioyxgq 好的，多谢指正。

konioyxgq commented 5 years ago

@CaptainEven 作者你好，我还想问一个关于BCNN的问题。我做了实验发现加入BCNN后和之前的结果比较，几乎差不多。不知道作者做过类似的试验吗？实验结果是什么样的呢？

CaptainEven commented 5 years ago

@konioyxgq BCNN主要用于细粒度分类，效果应该跟具体数据集有关，具体情况应该需要做更多的对比试验。