whai362 / PVT

Official implementation of PVT series
Apache License 2.0
1.73k stars 246 forks source link

what is the difference between pvt here? #40

Closed mathshangw closed 3 years ago

mathshangw commented 3 years ago

https://github.com/whai362/PVT/blob/main/pvt.py

https://github.com/whai362/PVT/blob/main/detection/pvt.py

2- does 12 epochs for detection and 300 epochs for classification ?

whai362 commented 3 years ago

Hi,the first one is for classification. The second one is for detection. You could follows the command in readme to train/test models.