yolact icon indicating copy to clipboard operation
yolact copied to clipboard

low mAP when doing transfer learning on a custom dataset?

Open amaanda opened this issue 4 years ago • 3 comments

Hi! Thank you for sharing your work :)

I've been attempting to do transfer learning with a class that is not present on the coco dataset. I've read the issues in this repo and followed the instructions about creating a custom dataset and running transfer learning with yolact++. I changed the config.py file with the information about my dataset, used COCO Annotator to check my annotations and it all seemed fine.

However, I'm not sure if the results I'm getting so far are correct.

[ 50]   10550 || B: 0.378 | C: 0.735 | M: 0.717 | S: 0.065 | I: 0.043 | T: 1.939 || ETA: 6 days, 12:19:33 || timer: 0.647

Calculating mAP...

       |  all  |  .50  |  .55  |  .60  |  .65  |  .70  |  .75  |  .80  |  .85  |  .90  |  .95  |
-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+
   box | 24.89 | 30.36 | 30.36 | 30.28 | 30.15 | 29.78 | 29.35 | 28.27 | 24.70 | 14.16 |  1.44 |
  mask | 28.09 | 30.60 | 30.28 | 30.25 | 30.25 | 30.25 | 30.20 | 29.92 | 29.07 | 26.92 | 13.22 |
-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+


[204]   42430 || B: 0.380 | C: 0.751 | M: 0.656 | S: 0.065 | I: 0.032 | T: 1.883 || ETA: 7 days, 7:51:03 || timer: 0.763

Calculating mAP...

       |  all  |  .50  |  .55  |  .60  |  .65  |  .70  |  .75  |  .80  |  .85  |  .90  |  .95  |
-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+
   box | 24.50 | 30.48 | 30.45 | 30.43 | 30.01 | 30.01 | 29.61 | 27.86 | 22.88 | 12.74 |  0.58 |
  mask | 27.87 | 30.22 | 30.18 | 30.16 | 30.14 | 29.86 | 29.82 | 29.50 | 28.75 | 26.59 | 13.48 |
-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+


[420]   87140 || B: 0.377 | C: 0.683 | M: 0.573 | S: 0.052 | I: 0.030 | T: 1.715 || ETA: 8 days, 9:55:49 || timer: 0.801

Calculating mAP...

       |  all  |  .50  |  .55  |  .60  |  .65  |  .70  |  .75  |  .80  |  .85  |  .90  |  .95  |
-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+
   box | 24.03 | 30.63 | 30.63 | 30.63 | 30.62 | 30.59 | 29.85 | 27.90 | 20.63 |  8.16 |  0.64 |
  mask | 28.06 | 30.31 | 30.29 | 30.29 | 30.29 | 30.29 | 30.25 | 29.86 | 29.02 | 26.91 | 13.11 |
-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+-------+

Thanks for you help!

amaanda avatar Oct 22 '20 14:10 amaanda

@amaanda May I ask how many dataset you had and how long you trained to get this mAP? Did you change learning rate or max_iteration? I'm doing a similar stuff and my mAP gets stuck around 14 for both box and mask.

kidpaul94 avatar Mar 24 '21 15:03 kidpaul94

I'm facing the same issue. Did you find out what was wrong?

shivani21998 avatar Dec 13 '21 06:12 shivani21998

I have the same problem, I have 40 imgs for training and 20imgs for val,but I only have on class

xuanzhiliu avatar Nov 14 '22 07:11 xuanzhiliu