Spaces:
Running
Running
File size: 3,558 Bytes
32faf2b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
manual_prompts = {
'carpet': [
# prompts, filtered phrase
['black hole', 'carpet'],
['thread', 'carpet'],
['defect.', 'carpet'],
],
'grid': [
# prompts, filtered phrase
['irregular pattern', 'grid'],
['defect.', 'grid'],
],
'leather': [
['defect.', 'leather'],
],
'tile': [
['defect.', 'tile'],
],
'wood': [
['defect.', 'wood'],
],
'bottle': [
# prompts, filtered phrase
['broken part. contamination. white broken.', 'bottle'],
],
'cable': [
# prompts, filtered phrase
['crack. flawed golden wire. black hole.', 'cable'],
],
'capsule': [
['white crack. hole.', 'capsule'],
# ['hole on capsule', 'capsule']
],
'hazelnut': [
# prompts, filtered phrase
['white print. crack. thread.', 'hazelnut'],
],
'metal_nut': [
# prompts, filtered phrase
['blue defect. black defect. red defect. scratch.', 'nut'],
],
'pill': [
# prompts, filtered phrase
['red defect. yellow defect. blue defect. crack. scratch.', 'pill'],
],
'screw': [
['defect.', 'screw'],
],
'toothbrush': [
['defect.', 'toothbrush'],
],
'transistor': [
['defect.', 'transistor'],
],
'zipper': [
['crack. broken leather.', 'zipper']
]
}
property_prompts = {
'carpet': 'the image of carpet have 1 dissimilar carpet, with a maximum of 5 anomaly. The anomaly would not exceed 0.9 object area. ',
'grid': 'the image of grid have 1 dissimilar grid, with a maximum of 5 anomaly. The anomaly would not exceed 0.9 object area. ',
'leather': 'the image of leather have 1 dissimilar leather, with a maximum of 5 anomaly. The anomaly would not exceed 0.9 object area. ',
'tile': 'the image of tile have 1 dissimilar tile, with a maximum of 5 anomaly. The anomaly would not exceed 0.9 object area. ',
'wood': 'the image of wood have 1 dissimilar wood, with a maximum of 5 anomaly. The anomaly would not exceed 0.9 object area. ',
'bottle': 'the image of bottle have 1 dissimilar bottle, with a maximum of 5 anomaly. The anomaly would not exceed 0.3 object area. ',
'cable': 'the image of cable have 1 dissimilar cable, with a maximum of 5 anomaly. The anomaly would not exceed 0.9 object area. ',
'capsule': 'the image of capsule have 1 dissimilar capsule, with a maximum of 5 anomaly. The anomaly would not exceed 0.6 object area. ',
'hazelnut': 'the image of hazelnut have 1 dissimilar hazelnut, with a maximum of 5 anomaly. The anomaly would not exceed 0.9 object area. ',
'metal_nut': 'the image of metal_nut have 1 dissimilar metal_nut, with a maximum of 5 anomaly. The anomaly would not exceed 1. object area. ',
'pill': 'the image of pill have 1 dissimilar pill, with a maximum of 5 anomaly. The anomaly would not exceed 1. object area. ',
'screw': 'the image of screw have 1 dissimilar screw, with a maximum of 5 anomaly. The anomaly would not exceed 0.1 object area. ',
'toothbrush': 'the image of toothbrush have 1 dissimilar toothbrush, with a maximum of 5 anomaly. The anomaly would not exceed 0.5 object area. ',
'transistor': 'the image of transistor have 1 dissimilar transistor, with a maximum of 5 anomaly. The anomaly would not exceed 1. object area. ',
'zipper': 'the image of zipper have 1 dissimilar zipper, with a maximum of 5 anomaly. The anomaly would not exceed 0.5 object area. ',
}
|