Pelee.Pytorch icon indicating copy to clipboard operation
Pelee.Pytorch copied to clipboard

fresh man ask for help about the dataset

Open OliverHxh opened this issue 4 years ago • 16 comments

hi~~~ i am a new man in this. your repository is amazing for me, but i have a problem with the dataset.maybe its not a big problem for u. i already download the Pelee_VOC.pth and put it to weights folder, and i follow the steps to download the dataset VOC2007, and it looks like this. image and i run the command python demo.py --dataset VOC\VOC2007 --config ./configs/Pelee_VOC.py --trained_model ./weights/Pelee_VOC.pth --show but it comes out the image i dont whats wrong? could u help me? thx~~~

OliverHxh avatar Jul 17 '19 13:07 OliverHxh

em.... worong folder path i was so stupid...

OliverHxh avatar Jul 18 '19 02:07 OliverHxh

@Jaychoukobbe Hi! I also meet a similar KeyError. Could you tell me what the right way to set folder path is? Currently I put VOC data at home: ~/data/VOCdevkit and ~/coco.

Z-Z-Han avatar Jul 27 '19 01:07 Z-Z-Han

@Z-Z-Han if u wanto train on VOC, just use VOC is ok , because in Pelee_VOC.py file, image this shows the dataset path, that all u need to do is to tell it what dataset u wanna train on VOC or COCO

OliverHxh avatar Jul 27 '19 02:07 OliverHxh

@Jaychoukobbe Get it! Thank you very very much!

Z-Z-Han avatar Jul 27 '19 02:07 Z-Z-Han

@Z-Z-Han i am still working on this codebase, if u have any question, u are welcome to ask...

OliverHxh avatar Jul 27 '19 02:07 OliverHxh

@Jaychoukobbe @Z-Z-Han @yxlijun did you train your own dataset? or train coco/voc dataset by your self? did you success with this code? and how long had you been training??

zhuixunforever avatar Jul 29 '19 03:07 zhuixunforever

@zhuixunforever i replace backbone with shuffleNetV2 and i train on VOC, i use 1080i gpu and almost train 3days, 400 epoch. by the way my batchsize is 64. the final MAP result is 0.62 when i use shufflenetV2 1X, and 0.67 when i use shufflenetV2 2X

OliverHxh avatar Aug 01 '19 08:08 OliverHxh

@zhuixunforever I trained the model on my dataset with over one hundred images,using 1080 gpu. The time for training is about on day, and I only change batchsize to 32. The performance is a little worse than ssd with vgg as backbone, but also acceptable. For an image with size 640*480, the inferencing time is about 1~2 seconds when using cpu.

Z-Z-Han avatar Aug 07 '19 08:08 Z-Z-Han

@Z-Z-Han what dataset u train on? and waht map u get?

OliverHxh avatar Aug 07 '19 09:08 OliverHxh

My own dataset

---Original--- From: "Jaychoukobbe"[email protected] Date: Wed, Aug 7, 2019 17:17 PM To: "yxlijun/Pelee.Pytorch"[email protected]; Cc: "Z-Z-Han"[email protected];"Mention"[email protected]; Subject: Re: [yxlijun/Pelee.Pytorch] fresh man ask for help about the dataset (#4)

@Z-Z-Han what dataset u train on? and waht map u get?

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub, or mute the thread.

Z-Z-Han avatar Aug 07 '19 09:08 Z-Z-Han

Not public available

---Original--- From: "Jaychoukobbe"[email protected] Date: Wed, Aug 7, 2019 17:17 PM To: "yxlijun/Pelee.Pytorch"[email protected]; Cc: "Z-Z-Han"[email protected];"Mention"[email protected]; Subject: Re: [yxlijun/Pelee.Pytorch] fresh man ask for help about the dataset (#4)

@Z-Z-Han what dataset u train on? and waht map u get?

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub, or mute the thread.

Z-Z-Han avatar Aug 07 '19 09:08 Z-Z-Han

@Z-Z-Han have ever tried on public dataset , like voc or coco?

OliverHxh avatar Aug 07 '19 09:08 OliverHxh

Not yet

---Original--- From: "Jaychoukobbe"[email protected] Date: Wed, Aug 7, 2019 17:21 PM To: "yxlijun/Pelee.Pytorch"[email protected]; Cc: "Z-Z-Han"[email protected];"Mention"[email protected]; Subject: Re: [yxlijun/Pelee.Pytorch] fresh man ask for help about the dataset (#4)

@Z-Z-Han have ever tried on public dataset , like voc or coco?

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub, or mute the thread.

Z-Z-Han avatar Aug 07 '19 09:08 Z-Z-Han

@Z-Z-Han all right

OliverHxh avatar Aug 07 '19 09:08 OliverHxh

@zhuixunforever I trained the model on my dataset with over one hundred images,using 1080 gpu. The time for training is about on day, and I only change batchsize to 32. The performance is a little worse than ssd with vgg as backbone, but also acceptable. For an image with size 640*480, the inferencing time is about 1~2 seconds when using cpu.

hao can i running it on cup,can you help me

duducode avatar Mar 16 '20 02:03 duducode

@Z-Z-Han i am still working on this codebase, if u have any question, u are welcome to ask...

i have following error "ImportError: No module named cpu_nms" how did you solve this.

Rafay2919 avatar Nov 12 '20 11:11 Rafay2919