Giter Site home page Giter Site logo

captaineven / repnet-mdnet-vehiclereid Goto Github PK

View Code? Open in Web Editor NEW
241.0 19.0 59.0 2.99 MB

Implementing RepNet(a two-stream multitask learning network) to do vehicle Re-identification, vehicle search(or vehicle match) with PyTorch 可用于车辆细粒度识别,车辆再识别,车辆匹配,车辆检索,RepNet/MDNet的一种PyTorch实现

License: MIT License

Python 100.00%
vehicle reid matching fine-grained-classification fine-grained-recognition vehicle-search vehicle-reidentification

repnet-mdnet-vehiclereid's Issues

related to test

如果没理解错的话,代码里仅对同时有model和color的数据集进行了train和test,那请问下test的时候有跑过整个test dataset(包括没有model或color)的实验嘛

Positive 和 Negative Set 问题

你好,我看到论文上面有Positive 和 Negative 训练集,但是在代码中貌似没有看到这个**的体现,代码是不是没有完全按照论文的方法来实现??

汽车品牌型号

你好,请问你有做过汽车品牌型号分类和识别吗?例如奥迪有Q5、Q3系列,通过车的图片来识别车的品牌具体型号

demo

你好,要测一张无标签的图片请问需要改哪些模块呀,好像得改网络模型是吗?

两个模型能整合吗

我看你弄了两个github项目。
RepNet-MDNet-VehicleReID用于识别车辆品牌、颜色。
Vehicle-Car-detection-and-multilabel-classification用于识别车辆朝向、车辆类型、颜色。
这两个模型能整合在一起,做成一个吗?,直接能识别车辆品牌、颜色、车辆朝向、车辆类型这些车辆属性。

车辆检索和车辆重识别

你好,请问车辆检索和车辆重识别是同一个问题吗?然后可以麻烦你写一下具体实现步骤吗?谢谢!

其他数据集测试

你好,我是用其他数据集测试的是时候就一点也不准,这是因为数据样本分布问题吗?

关于网络输入输出的维度

想请教一下,初始化网络的时候这两个参数 “out_ids=10086” 和 “out_attribs=257” 分别代表什么呢?这两个取值是如何确定的?如果要新增自己需要识别的车辆品牌,是否需要修改这两个值?
tmp

求train好的model

由于计算资源有限,所以请问有最后完全train好的model可以提供下载嘛

pair_set_car.txt 和 CarReIDCrop 能分享一下吗

你好,在测试特征编码相似度的时候,发现缺少了pair_set_car.txt 和 CarReIDCrop。
test_car_match_data(resume='epoch_14.pth',
pair_set_txt='/mnt/diskc/even/Car_DR/ArcFace_pytorch/data/pair_set_car.txt',
img_root='/mnt/diskc/even/CarReIDCrop', # CarReID_data
batch_size=16)
能分享一下吗?感谢。

如何训练。

您好,请问数据源和模型都下载以后,应如何进行训练呢?

训练自己的数据集,出现问题

请问我基于你的代码和流程做自己的数据集,总共9万多张image,400条boat,200多个model,训练一开始报如下错误,请教一下会是什么原因呢?
Training... => Epoch Train loss Train acc Test acc /pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:105: void cunn_ClassNLLCriterion_updateOutput_kernel(Dtype *, Dtype *, Dtype *, long *, Dtype *, int, int, int, int, long) [with Dtype = float, Acctype = float]: block: [0,0,0], thread: [12,0,0] Assertion t >= 0 && t < n_classesfailed. /pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:105: void cunn_ClassNLLCriterion_updateOutput_kernel(Dtype *, Dtype *, Dtype *, long *, Dtype *, int, int, int, int, long) [with Dtype = float, Acctype = float]: block: [0,0,0], thread: [15,0,0] Assertiont >= 0 && t < n_classesfailed. Traceback (most recent call last): File "RepNet.py", line 1606, in <module> train(resume=None) # 从头开始训练 File "RepNet.py", line 1541, in train epoch_loss.append(loss.cpu().item()) RuntimeError: CUDA error: device-side assert triggered

accuracy =0

您好,请问在训练到iteration=100时,accuracy=0是正常现象吗,loss在39到40之间
用的函数是train(resume=None)
感谢您的回复

关于车辆图片的特征提取

作者您好,我想问一下,这个网络可以将视角差距较大的车辆图片识别成同一辆车吗?我用VehicleID里的图片测试了一下,取vid相同的两张图片,分别用网络的branch5提取出1024维特征向量,经过L2归一化之后计算欧式距离。如果都是正面的视角,提取出的特征向量欧氏距离较小,如果一个是正面一个是背面,欧氏距离就比较大了。由于正面和背面的图片视角差距较大,似乎很难找到相同的特征点,不知道这个网络能不能识别?另外,在训练的时候,需要将正面和背面的图片分开,当成是vid不同的两辆车吗?
show

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.